web-dev-qa-db-fra.com

Pandas Écriture de colonnes de trame de données dans csv

J'écris un script pour réduire un gros fichier .xlsx avec des en-têtes dans un csv, puis j'écris un nouveau fichier csv avec uniquement les colonnes requises en fonction du nom de l'en-tête.

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

L'erreur que je reçois est avec ce dernier morceau de code, et il dit

ValueError: Writing 102 cols but got 4 aliases

Je suis sûr que je néglige quelque chose de stupide, mais j'ai lu la documentation de to_csv sur le site Web pandas et je suis toujours perdu. Je sais que j'utilise le to_csv paramètres, mais je n'arrive pas à comprendre la documentation, je suppose.

Toute aide est la bienvenue, merci!

24
Harrison Boles

La façon de sélectionner des colonnes spécifiques est la suivante -

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