pandas - 如何绘制熊猫数据框的 2 列，不包括第三列选择的行

Question

我有一个看起来像这样的熊猫数据框：

       Nstrike     Nprem  TDays
0    0.920923  0.000123    2
1    0.951621  0.000246    2
2    0.957760  0.001105    2
..............................
16   0.583251  0.000491    7
17   0.613949  0.000614    7
18   0.675344  0.000368    7
..............................
100  1.013016  0.029592   27
101  1.043713  0.049730   27
102  1.074411  0.071218   27
etc.

我想在 col.3 选择的单独图中绘制 col.1 与 col.2 的图表，甚至可能使用不同的颜色。

我能看到的唯一方法是将数据帧分成每个 col.3 值的离散数据帧。

或者我可以放弃 pandas，只将 col.3 子集制作成普通的 python 数组。

如果可以简化问题，我可以自由更改数据框的结构。

标签： pandasdataframematplotlib

Answer 1

IIUC，您可以将其用作骨架，并根据需要对其进行自定义：

for g, data in df.groupby('TDays'):
    plt.plot(data.Nstrike, data.Nprem, label='TDays '+str(g))
    plt.legend()
    plt.savefig('plot_'+str(g)+'.png')
    plt.close()

您的第一个情节将如下所示：

你的第二个：

等等