绘制一组熊猫数据框

时间:2018-07-16 18:25:34

标签: python pandas

我有一些大熊猫DataFrame,可以将其视为时间点。

它们看起来像这样:

时间1:

pd.DataFrame({'type':[1,2],'b':[200,400]})

    b   type
0   200 1
1   400 2

时间2:

pd.DataFrame({'type':[1,4],'b':[100,300]})

    b   type
0   100 1
1   300 4

我正在尝试将DataFrame组合在一起并绘制它们。我认为最好将其绘制为折线图,其中数据按type列分组,而b列用作y轴值。

1 个答案:

答案 0 :(得分:1)

我建议使用散点图,将类型列上的值分隔为不同的颜色,并沿x轴绘制。

要合并数据帧,可以使用join。假设您的第一个df为a,第二个为b

df = a.join(b.set_index('b'), on='b')
df.index.name = 'index'  # for plotting as x-axis

然后您可以绘制散点图:

cmap = {0: 'black', 1: 'orchid', 2: 'seagreen', 3: ...}
colors = np.array([cmap[c] for c in df.type])
fig, ax = plt.subplots()

df.plot.scatter('index', 'type', colors=colors, ax=ax)
plt.show()

您当然可以选择绘制线条图-让我知道这是否真的是您想要的,并且您不能从上面的示例中弄清楚。