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

时间:2018-10-12 21:48:45

标签: pandas dataframe matplotlib

我有一个熊猫数据框,如下所示:

       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.

我想在第3列选择的单独绘图中绘制第1列与第2列的图形,甚至可能使用不同的颜色。

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

或者我可以放弃熊猫,而只是将col.3子集变成普通的python数组。

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

1 个答案:

答案 0 :(得分: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()

您的第一个情节如下:

enter image description here

您的第二个:

enter image description here

依此类推