Pandy: jak szybko przekonwertować kolumnę na listę
Możesz użyć jednej z następujących metod, aby przekonwertować kolumnę ramki DataFrame pandy na listę:
Metoda 1: Użyj tolist()
df[' my_column ']. tolist ()
Metoda 2: Użyj listy()
list(df[' my_column '])
Obie metody zwrócą dokładnie ten sam wynik.
Poniższe przykłady pokazują, jak używać każdej z tych metod z następującą ramką DataFrame pand:
import pandas as pd #createDataFrame df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'B', 'B', 'B'], ' points ': [99, 90, 93, 86, 88, 82], ' assists ': [33, 28, 31, 39, 34, 30]}) #view DataFrame print (df) team points assists 0 to 99 33 1 A 90 28 2 A 93 31 3 B 86 39 4 B 88 34 5 B 82 30
Metoda 1: Konwertuj kolumnę na listę za pomocą tolist()
Poniższy kod pokazuje, jak używać funkcji tolist() do konwersji kolumny „punkty” ramki DataFrame na listę:
#convert column to list my_list = df[' points ']. tolist () #view list print (my_list) [99, 90, 93, 86, 88, 82]
Możemy potwierdzić, że wynikiem jest lista, używając funkcji type() :
#check datatype type (my_list) list
Metoda 2: Konwertuj kolumnę na listę za pomocą list()
Poniższy kod pokazuje, jak używać funkcji list() do konwersji kolumny „punkty” ramki DataFrame na listę:
#convert column to list my_list = list(df[' points ']) #view list print (my_list) [99, 90, 93, 86, 88, 82]
Możemy potwierdzić, że wynikiem jest lista, używając funkcji type() :
#check data type type (my_list) list
Należy pamiętać, że obie metody zwracają dokładnie takie same wyniki.
Należy pamiętać, że w przypadku bardzo dużych ramek danych metoda tolist() jest zazwyczaj najszybsza.
Dodatkowe zasoby
Poniższe samouczki wyjaśniają, jak wykonywać inne typowe funkcje z kolumnami ramki DataFrame pandy:
Jak usunąć kolumny w Pandach
Jak wykluczyć kolumny w Pandach
Jak zastosować funkcję do wybranych kolumn w Pandach
Jak zmienić kolejność kolumn w Pandas DataFrame