重命名不同文件的同一列

时间:2018-10-05 16:35:47

标签: python-3.x pandas

我正在用python编写代码。我通过合并多个相同名称的文件创建了一个输出csv文件。现在我要绘制它。但是我在绘图过程中面临一个问题,就是绘图所需的列对于不同的输出文件具有不同的名称。我的代码是:

df = pd.read_csv('out.csv')
if df.columns[1] == 'User Time[0]':
  df.rename(columns = {'User Time[0]' : 'User Time'})
elif df.columns[1] == 'User Time[0]':
  df.rename(columns = {'User Time[1]' : 'User Time'})
elif df.columns[1] == 'User Time[0]':
  df.rename(columns = {'User Time[2]' : 'User Time'})
else:
  df.columns[1] == 'User Time'
df.plot(x='User Time', y='GDOP[2]', ax=ax, legend=False)

plt.ylim([0.0, 20.0])
plt.show()

与代码中一样,我希望第二列名称为“用户时间”。 请帮助。谢谢。

1 个答案:

答案 0 :(得分:1)

您可以使用以下代码:

df = pd.read_csv('out.csv')
clm = list(df.columns)
clm[1] = 'User Time'
df.columns = clm

df.plot(x='User Time', y='GDOP[2]', ax=ax, legend=False)

plt.ylim([0.0, 20.0])
plt.show()
相关问题