编写python程序,找到一批金融数据,根据某个字段分类,画出不同颜色的散点图
时间: 2023-06-25 08:02:56 浏览: 187
以下是一个示例程序,可以根据输入的数据文件和分类字段,生成对应的散点图:
```python
import pandas as pd
import matplotlib.pyplot as plt
# 读取数据文件
data = pd.read_csv('data.csv')
# 根据分类字段分组
groups = data.groupby('category')
# 遍历每个分组,并绘制散点图
fig, ax = plt.subplots()
for name, group in groups:
ax.plot(group['x'], group['y'], marker='o', linestyle='', ms=6, label=name)
ax.legend()
plt.show()
```
其中,`data.csv` 是数据文件,包含两个字段 `x` 和 `y`,以及一个分类字段 `category`。可以根据实际数据格式进行修改。
运行程序后,将生成一个散点图,每个分类对应不同的颜色和标记。可以根据需要进行进一步的美化和调整。
阅读全文