'Usecols' funktioniert nicht beim Schreiben auf CSV

Question

Ich habe Tow CSV-Dateien File1 und File2. Ich modifiziere File2 im Grunde so, dass es mit den Spalten in File1 übereinstimmt. Im letzten Schritt muss ich das modifizierte File2 abschneiden, um nur die ersten 50 Spalten zu haben. Ich habe dieses Programm geschrieben.

import pandas as pd 
df1 = pd.read_csv('C:/Users/merged_34_final_ordered.csv', dtype = 'unicode') 
df2 = pd.read_csv('C:/Users/merged_52_raw_ip_2016_06_30.csv', dtype = 'unicode') 
df3 = df2.T.reindex(df1.columns).T 
df3['idSite'] = pd.Series([52 for x in range(len(df3.index))], index=df3.index) 
df3.to_csv('C:/Users/merged_52_raw_ip_2016_06_30_reordered.csv',usecols = list(range(50)),index=False) 

Dies scheint nicht zu funktionieren, da es die Datei nicht abschneidet. Es funktioniert nur, wenn wir es während der read_csv hinzufügen.

Answer 1

Das richtige Argument für DataFrame.to_csv() ist columns, nicht usecols — letztere für pd.read_csv() ist. Wie @ptrj jedoch angibt, erwarten Sie to_csv() Spaltennamen. Sie können die ersten 50 Spalten auswählen und to_csv() auf sie nennen,

df.iloc[:,:50].to_csv(filename, index=False) 

oder Verwendung @ ptrj Ansatz die Namen der ersten 50 Spalten zu erhalten.

Answer 2

können Einige Ihrer Code vereinfacht werden:

df1 = pd.read_csv(...) 
df2 = pd.read_csv(...) 

df3 = df2.reindex(columns=df1.columns) 
df3['idSite'] = 52 

# names of the first 50 cols: 
cols = df3.columns[:50] 
df3.to_csv('filename', columns=cols, index=False) 

Sie erhalten den Fehler, da to_csv Spaltennamen erwartet.