python - 如何将数据框的特定列写入 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)