python - 绘制一组 pandas 数据框

Question

我有一些 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 轴值。

Answer 1

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

要组合数据帧，您可以使用 join 。假设您的第一个 df 是 a，第二个 df 是 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()

您当然可以选择绘制线图 - 请告诉我这是否真的是您想要的，并且您无法从上面的示例中找出答案。