python pandas 有两列数,先用第一列数cut分组,再求每个分组的累计值和第二列数据的累积值,然后做折线图
时间: 2023-01-11 10:55:36 浏览: 53
使用Python画折线图
你可以先使用pandas的cut函数将第一列的数据分组,然后使用groupby函数按照分组进行分组,然后使用cumsum函数求出每个分组的累计值。这样就可以得到每个分组的累计值了。对于第二列的数据的累积值,可以使用pandas的cumsum函数求出累计值。
然后,可以使用matplotlib绘制折线图,将分组的累计值和第二列数据的累计值分别作为折线图的x轴和y轴。具体代码如下:
```
import pandas as pd
import matplotlib.pyplot as plt
# 读入数据
df = pd.read_csv('data.csv')
# 将第一列的数据分组
bins = [0, 10, 20, 30]
labels = ['0-10', '10-20', '20-30']
df['group'] = pd.cut(df['col1'], bins=bins, labels=labels)
# 按照分组进行分组,并求出每个分组的累计值
df_grouped = df.groupby('group').cumsum()
# 求出第二列数据的累计值
df['col2_cumsum'] = df['col2'].cumsum()
# 绘制折线图
plt.plot(df_grouped['col1'], df['col2_cumsum'])
plt.show()
```
阅读全文