Pandas: cara cepat mengubah kolom menjadi daftar

Anda dapat menggunakan salah satu metode berikut untuk mengonversi kolom pandas DataFrame menjadi daftar:

Metode 1: Gunakan daftar()

 df[' my_column ']. tolist ()

Metode 2: Gunakan daftar()

 list(df[' my_column '])

Kedua metode akan memberikan hasil yang sama persis.

Contoh berikut menunjukkan cara menggunakan masing-masing metode ini dengan pandas DataFrame berikut:

 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

Metode 1: Ubah Kolom menjadi Daftar Menggunakan tolist()

Kode berikut menunjukkan cara menggunakan fungsi tolist() untuk mengonversi kolom “poin” DataFrame menjadi daftar:

 #convert column to list
my_list = df[' points ']. tolist ()

#view list
print (my_list)

[99, 90, 93, 86, 88, 82]

Kami dapat mengonfirmasi bahwa hasilnya adalah daftar menggunakan fungsi type() :

 #check datatype
type (my_list)

list

Metode 2: Ubah Kolom menjadi Daftar Menggunakan list()

Kode berikut menunjukkan cara menggunakan fungsi list() untuk mengonversi kolom “poin” DataFrame menjadi daftar:

 #convert column to list
my_list = list(df[' points '])
#view list
print (my_list)

[99, 90, 93, 86, 88, 82]

Kami dapat mengonfirmasi bahwa hasilnya adalah daftar menggunakan fungsi type() :

 #check data type
type (my_list)

list

Perhatikan bahwa kedua metode memberikan hasil yang persis sama.

Perhatikan bahwa untuk DataFrame yang sangat besar, metode tolist() cenderung menjadi yang tercepat.

Sumber daya tambahan

Tutorial berikut menjelaskan cara melakukan fungsi umum lainnya dengan kolom pandas DataFrame:

Cara menghapus kolom di Pandas
Cara mengecualikan kolom di Pandas
Cara menerapkan fungsi ke kolom yang dipilih di Pandas
Bagaimana mengubah urutan kolom di Pandas DataFrame