pandas 和 matplotlib 绘图数据框

Question

我有以下熊猫数据框

test = pd.DataFrame({'cities':['A','B','A','B','A','B',],
               'date':['2016-9-1','2016-9-1','2016-9-2','2016-9-2','2016-9-3','2016-9-3'],
               'count':[10,20,30,40,50,60]})

我想绘制两个散点图，一个用于城市“A”，一个用于城市“B”。在每个图中，x 轴是日期，y 轴是计数。

我试了一下，但代码太丑了，我不敢张贴。希望有人能以优雅的方式提供帮助。

谢谢。

Answer 1

设置

test = pd.DataFrame({'cities':['A','B','A','B','A','B',],
               'date':['2016-9-1','2016-9-1','2016-9-2','2016-9-2','2016-9-3','2016-9-3'],
               'count':[10,20,30,40,50,60]})

test.date = pd.to_datetime(test.date)

解决方案

import matplotlib.pyplot as plt


gb = test.groupby('cities')

fig, axes = plt.subplots(gb.ngroups)
for i, (name, df) in enumerate(gb):
    ax = axes[i]
    ax.scatter(df['date'].values, df['count'].values)
    ax.set_title(name)

fig.autofmt_xdate()
fig.tight_layout()