Pandas将数据帧列写入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

我确定我忽略了一些愚蠢的东西,但我已经阅读了熊猫网站上的to_csv文档,我仍然不知所措.我知道我正在使用to_csv参数不正确,但我似乎无法理解我的文档.

任何帮助表示赞赏,谢谢!

Answer 1

选择特定列的方法是这样的 -

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