使用 pandas 導出 csv 時指定第一列

Question

想知道如何使用 pandas 指定第一列？

例如，我有一個帶有變量var1, var2, var3,...varx的 dataframe df ，我希望生成的 csv 中的第一列是var3 。

df.to_csv('location/export.csv', index=False)

Answer 1

您可以通過執行以任何方式重新排列列

df[['var3', 'var1', 'var2', ..., 'varx']].to_csv('location/export.csv', index=False)

如果您只想在開始時使用 var3，則可以使用較少手動的方式

cols = df.columns.tolist()
cols.remove('var3')
df[['var3'] + cols].to_csv('location/export.csv', index=False)