如何將數據框的特定列寫入 CSV？

Question

我正在編寫一個腳本來將帶有標題的大型 .xlsx 文件縮減為 CSV，然后根據標題名稱編寫一個僅包含所需列的新 CSV 文件。

import pandas
import csv

df = pandas.read_csv('C:\\Python27\\Work\\spoofing.csv')

time = df["InviteTime (Oracle)"]
orignum = df["Orig Number"]
origip = df["Orig IP Address"]
destnum = df["Dest Number"]

df.to_csv('output.csv', header=[time,orignum,origip,destnum])

我得到的錯誤是最后一段代碼，它說

ValueError: Writing 102 cols but got 4 aliases

我確定我忽略了一些愚蠢的事情，但我已經閱讀了 pandas 網站上的to_csv文檔，但我仍然不知所措。 我知道我在濫用 to_csv 參數，但我似乎無法理解文檔。

任何幫助表示贊賞，謝謝！

Answer 1

選擇特定列的方法是這樣的 -

header = ["InviteTime (Oracle)", "Orig Number", "Orig IP Address", "Dest Number"]
df.to_csv('output.csv', columns = header)

Answer 2

column_list=["column_name1", "column_name2", "column_name3", "column_name4"]

#filter the dataframe beforehand
ds[column_list].to_csv('output.csv',index=False)

#or use columns arg
ds.to_csv('output.csv', columns = column_list,index=False)

我提供index=False arg 以便僅寫入列值