提问者:小点点

熊猫将数据帧列写入csv


我正在编写一个脚本,将一个带有标题的大型. 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参数,但我想我似乎无法理解文档。

感谢您的帮助,谢谢!


共1个答案

匿名用户

选择特定列的方法如下-

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