Ustaw kolejność kolumn w ramce danych pandy

107

Czy istnieje sposób na zmianę kolejności kolumn w pandas dataframe w oparciu o moje osobiste preferencje (tj. Nie posortowane alfabetycznie lub numerycznie, ale bardziej jak przestrzeganie pewnych konwencji)?

Prosty przykład:

frame = pd.DataFrame({
        'one thing':[1,2,3,4],
        'second thing':[0.1,0.2,1,2],
        'other thing':['a','e','i','o']})

daje to:

   one thing other thing  second thing
0          1           a           0.1
1          2           e           0.2
2          3           i           1.0
3          4           o           2.0

Ale zamiast tego chciałbym to:

   one thing second thing  other thing
0          1           0.1           a
1          2           0.2           e
2          3           1.0           i
3          4           2.0           o

(Proszę podać ogólne rozwiązanie, a nie specyficzne dla tego przypadku. Wielkie dzięki.)

durbachit
źródło

Odpowiedzi:

162

Po prostu wybierz zamówienie samodzielnie, wpisując nazwy kolumn. Zwróć uwagę na podwójne nawiasy:

frame = frame[['column I want first', 'column I want second'...etc.]]
A.Kot
źródło
28
Działa to tylko z tym raczej małym przykładem. Jeśli wczytujesz dane z innego źródła, takiego jak plik csv lub tabela bazy danych, nie możesz użyć tej odpowiedzi. A te wydają się być znacznie częstsze. OP zażądał rozwiązania ogólnego.
chrisfs
85

Możesz użyć tego:

columnsTitles = ['onething', 'secondthing', 'otherthing']

frame = frame.reindex(columns=columnsTitles)
Okroshiashvili
źródło
6
Chociaż większość innych rozwiązań jest bardziej zwięzła, uważam to za najbardziej czytelne dla każdego, kto nie jest w 100% zaznajomiony pandas.
Dirk,
3
Pamiętaj jednak, aby przypisać wartość zwracaną do zmiennej, nie zmienia to kolejności kolumn w miejscu (przynajmniej nie w wersji pandas0.23).
Dirk,
Dzięki @Dirk za sugestię
Okroshiashvili
34

Oto rozwiązanie, z którego korzystam bardzo często. Gdy masz duży zestaw danych z tonami kolumn, zdecydowanie nie chcesz ręcznie zmieniać kolejności wszystkich kolumn.

To, co możesz i najprawdopodobniej chcesz zrobić, to po prostu zamówić kilka pierwszych kolumn, których często używasz, i pozwolić wszystkim pozostałym kolumnom być sobą. Jest to powszechne podejście w R.df %>%select(one, two, three, everything())

Możesz więc najpierw ręcznie wpisać kolumny, które chcesz uporządkować, i umieścić je przed wszystkimi innymi kolumnami na liście cols_to_order.

Następnie tworzysz listę dla nowych kolumn, łącząc pozostałe kolumny:

new_columns = cols_to_order + (frame.columns.drop(cols_to_order).tolist())

Następnie możesz użyć new_columnsinnych sugerowanych rozwiązań.

import pandas as pd
frame = pd.DataFrame({
    'one thing': [1, 2, 3, 4],
    'other thing': ['a', 'e', 'i', 'o'],
    'more things': ['a', 'e', 'i', 'o'],
    'second thing': [0.1, 0.2, 1, 2],
})

cols_to_order = ['one thing', 'second thing']
new_columns = cols_to_order + (frame.columns.drop(cols_to_order).tolist())
frame = frame[new_columns]

   one thing  second thing other thing more things
0          1           0.1           a           a
1          2           0.2           e           e
2          3           1.0           i           i
3          4           2.0           o           o
Lala La
źródło
1
genialny, doskonały. dziękuję za powstrzymanie mnie przed wpisywaniem nazwy lub indeksu każdej kolumny
stuart
To jest ogólna odpowiedź i powinna być akceptowaną odpowiedzią
CarlosH
26

Możesz też zrobić coś takiego df = df[['x', 'y', 'a', 'b']]

import pandas as pd
frame = pd.DataFrame({'one thing':[1,2,3,4],'second thing':[0.1,0.2,1,2],'other thing':['a','e','i','o']})
frame = frame[['second thing', 'other thing', 'one thing']]
print frame
   second thing other thing  one thing
0           0.1           a          1
1           0.2           e          2
2           1.0           i          3
3           2.0           o          4

Możesz również uzyskać listę kolumn za pomocą:

cols = list(df.columns.values)

Wynik da coś takiego:

['x', 'y', 'a', 'b']

Które można następnie łatwo zmienić ręcznie.

omri_saadon
źródło
13

Skonstruuj go za pomocą listy zamiast słownika

frame = pd.DataFrame([
        [1, .1, 'a'],
        [2, .2, 'e'],
        [3,  1, 'i'],
        [4,  4, 'o']
    ], columns=['one thing', 'second thing', 'other thing'])

frame

   one thing  second thing other thing
0          1           0.1           a
1          2           0.2           e
2          3           1.0           i
3          4           4.0           o
piRSquared
źródło
Nie udało mi się uzyskać „nazwy kolumny”: danych do pracy na liście, tak jak w przypadku dyktowania.
Kim Miller
10

Możesz także użyć OrderedDict:

In [183]: from collections import OrderedDict

In [184]: data = OrderedDict()

In [185]: data['one thing'] = [1,2,3,4]

In [186]: data['second thing'] = [0.1,0.2,1,2]

In [187]: data['other thing'] = ['a','e','i','o']

In [188]: frame = pd.DataFrame(data)

In [189]: frame
Out[189]:
   one thing  second thing other thing
0          1           0.1           a
1          2           0.2           e
2          3           1.0           i
3          4           2.0           o
MaxU
źródło
6

Dodaj parametr „kolumny”:

frame = pd.DataFrame({
        'one thing':[1,2,3,4],
        'second thing':[0.1,0.2,1,2],
        'other thing':['a','e','i','o']},
        columns=['one thing', 'second thing', 'other thing']
)
Irene
źródło
4

Spróbuj indeksować (więc chcesz mieć ogólne rozwiązanie nie tylko do tego, aby kolejność indeksowania była dokładnie taka, jak chcesz):

l=[0,2,1] # index order
frame=frame[[frame.columns[i] for i in l]]

Teraz:

print(frame)

Jest:

   one thing second thing  other thing
0          1           0.1           a
1          2           0.2           e
2          3           1.0           i
3          4           2.0           o
U10-Forward
źródło
-2

Uważam, że jest to najbardziej proste i działające:

df = pd.DataFrame({
        'one thing':[1,2,3,4],
        'second thing':[0.1,0.2,1,2],
        'other thing':['a','e','i','o']})

df = df[['one thing','second thing', 'other thing']]
Sando K.
źródło