Zachowanie porządku kolumn w Pythonie Pandy DataFrame

Question

Czy istnieje sposób na zachowanie kolejności kolumn w pliku csv podczas czytania i pisania za pomocą Pand Python? Na przykład w tym kodzie pliki wyjściowe mogą być inne, ponieważ kolumny nie są zachowywane.

Answer 1

Wydaje się, że błąd w aktualnej wersji Pandy ('0.11.0'), który oznacza, że ​​odpowiedź Mattiego Johna nie zadziała. Jeśli określisz kolumny do zapisywania do pliku, zostaną one zapisane w kolejności alfabetycznej, ale po prostu ponownie oznaczone etykietą zgodnie z listą w cols. Na przykład, ten kod:

import pandas 
dfdict={} 
dfdict["a"]=[1,2,3,4] 
dfdict["b"]=[5,6,7,8] 
dfdict["c"]=[9,10,11,12] 
df=pandas.DataFrame(dfdict) 
df.to_csv("dfTest.txt","\t",header=True,cols=["b","a","c"]) 

wyniki w tym (błędnej) Wyjście:

b a c 
0 1 5 9 
1 2 6 10 
2 3 7 11 
3 4 8 12 

Można sprawdzić, która wersja pandy zostały zainstalowane przez wykonanie:

pandas.version.version 

Dokumentacja to_csv to here

Wygląda na to, że jest to znany błąd i zostanie rozwiązany w kolejnej wersji (0.11.1):

https://github.com/pydata/pandas/issues/3489

UPDATE: Nadal nie została nowa wersja pandy, ale istnieje obejście opisane tutaj, który nie wymaga korzystania inna wersja pandy:

github.com/pydata/pandas/issues/3454

Więc zmianie ostatniego wiersza w bloku kodu powyżej do następujących będzie działać poprawnie:

df.to_csv("dfTest.txt","\t",header=True,cols=["b","a","c"], engine='python') 

UPDATE wydaje się, że argument "cols" został przemianowany na "columns", a argument "engine" jest przestarzały (nie jest już dostępny) w najnowszych wersjach pandy. Ten błąd został naprawiony w wersji 0.19.0.

Answer 2

Kolejność kolumn powinna być ogólnie zachowana podczas czytania, a następnie zapisywania pliku CSV, ale jeśli z jakiegoś powodu nie są one w żądanej kolejności, można użyć argumentu ze słowem kluczowym columns w to_csv.

Na przykład, jeśli masz CSV z kolumn A, B, C, D:

data = pd.read_csv(filename) 
data.to_csv(filename, columns=['a', 'b', 'c', 'd']) 

Answer 3

Innym rozwiązaniem jest to zrobić:

import pandas as pd 
data = pd.read_csv(filename) 
data2 = df[['A','B','C']] #put 'A' 'B' 'C' in the desired order 
data2.to_csv(filename)