How to beginner · 3 min read

How to use ChatGPT for data analysis

Q: How to use ChatGPT for data analysis

Use ChatGPT by sending your data analysis queries or datasets as prompts to the OpenAI API. Process responses to generate summaries, insights, or code snippets for data manipulation and visualization.

Quick answer

Use ChatGPT by sending your data analysis queries or datasets as prompts to the OpenAI API. Process responses to generate summaries, insights, or code snippets for data manipulation and visualization.

PREREQUISITES

Python 3.8+
OpenAI API key (free tier works)
pip install openai>=1.0

Setup

Install the openai Python package and set your API key as an environment variable for secure access.

bash

pip install openai

Step by step

This example shows how to send a data analysis question to ChatGPT using the gpt-4o model and print the response.

python

import os
from openai import OpenAI

client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])

prompt = (
    "I have a dataset with sales figures for Q1 and Q2. "
    "Can you help me analyze the trend and suggest visualization types?"
)

response = client.chat.completions.create(
    model="gpt-4o",
    messages=[{"role": "user", "content": prompt}]
)

print(response.choices[0].message.content)

output

The sales figures show an upward trend from Q1 to Q2. A line chart or bar chart would effectively visualize this growth over time. Consider adding a trendline to highlight the increase.

Common variations

You can use ChatGPT to generate Python code for data analysis libraries like pandas or matplotlib. Also, try streaming responses or use other models like claude-3-5-sonnet-20241022 for more detailed analysis.

python

import os
from openai import OpenAI

client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])

prompt = (
    "Generate Python code to plot sales data using matplotlib."
)

response = client.chat.completions.create(
    model="gpt-4o",
    messages=[{"role": "user", "content": prompt}]
)

print(response.choices[0].message.content)

output

import matplotlib.pyplot as plt

sales = [150, 200]
quarters = ['Q1', 'Q2']

plt.bar(quarters, sales)
plt.title('Sales by Quarter')
plt.show()

Troubleshooting

If you get incomplete answers, increase max_tokens in the API call.
For API authentication errors, verify your OPENAI_API_KEY environment variable is set correctly.
If responses are off-topic, refine your prompt with more context or examples.

✅

Key Takeaways

Use ChatGPT to generate insights and code snippets for data analysis tasks.
Always secure your API key using environment variables and the latest SDK patterns.
Refine prompts with clear context to get accurate and actionable data analysis responses.

Verified 2026-04 · gpt-4o, claude-3-5-sonnet-20241022

Verify ↗