Pandas将数据帧列写入csv

时间:2014-02-25 16:07:59

标签: python csv pandas

我正在编写脚本以将带有标头的大型.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 个答案:

答案 0 :(得分:50)

选择特定列的方法是 -

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