Pandas Writing Dataframe Columns to csv

Question

I'm writing a script to reduce a large .xlsx file with headers into a csv, and then write a new csv file with only the required columns based on header name.

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]) 

The error I'm getting is with that last bit of code, and it says

ValueError: Writing 102 cols but got 4 aliases 

I'm sure i'm overlooking something stupid, but I've read over the to_csv documentation on the pandas website and I'm still at a loss. I know I'm using the to_csv parameters incorrectly but I can't seem to get my head around the documentation I guess.

Any help is appreciated, thanks!

Answer 1

The way to select specific columns is this -

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