Question

我正在使用以下代码进行pandas plot。它接收一个文件，并为特定行（轨迹ID：即'loc.27404'）绘制图。

但是，此代码手动指定我要绘制的行。

我有另一个包含我要绘制的所有行（有100个）的文件 - 如何编写一个允许我输入此文件的脚本，以便使用下面的代码绘制这些特定行。我似乎无法写任何有意义的东西。

data = {}
for line in File:
    cols = line.strip().split('\t')
    vals = map(float,cols[6:])
    data[cols[3]] = vals

fig,ax = plt.subplots(4,figsize=[15,20])

l1= 'loc.27404'
l2= 'loc.37387'
l3 = 'loc.05134'
l4 = 'loc.10034'

pd.Series(data[l1],index=xticks).plot(ax=ax[0])
pd.Series(data[l2],index=xticks).plot(ax=ax[1])
pd.Series(data[l3],index=xticks).plot(ax=ax[2])
pd.Series(data[l4],index=xticks).plot(ax=ax[3])

Answer 1

一个潜在的解决方案是读取100行并将结果存储在列表中，例如：['loc.27404','loc.37387','loc.05134','loc.10034']用于for循环：

for i,line_to_plot in enumerate(['loc.27404','loc.37387','loc.05134','loc.10034']):
    pd.Series(data[line_to_plot],index=xticks).plot(ax=ax[i])

pandas plot - 在文件

1 个答案: