Bagaimana untuk Subset DataFrame Pandas Berdasarkan Senarai Nilai?

Mensubsetkan Bingkai Data Pandas Berdasarkan Senarai Nilai

Memilih baris dalam bingkai data Pandas berdasarkan nilai tertentu adalah mudah menggunakan pengendali kesamaan . Walau bagaimanapun, apabila berurusan dengan pelbagai nilai, pendekatan yang lebih fleksibel diperlukan. Artikel ini menerangkan cara untuk subset bingkai data menggunakan senarai nilai.

Masalah:

Pertimbangkan bingkai data berikut:

df = DataFrame({'A': [5,6,3,4], 'B': [1,2,3,5]})

df

     A   B
0    5   1
1    6   2
2    3   3
3    4   5

Kami mahu untuk memilih baris di mana lajur 'A' sepadan dengan nilai dalam senarai tertentu, seperti [3, 6]:

list_of_values = [3, 6]

y = df[df['A'] in list_of_values]

Penyelesaian:

Kaedah isin kerangka data menyediakan cara yang mudah untuk mencapai perkara ini:

df[df['A'].isin([3, 6])]

Ini mengembalikan baris berikut:

     A    B
1    6    2
2    3    3

Untuk pemilihan songsang, tidak termasuk baris dengan nilai dalam senarai yang diberikan, gunakan operator ~:

df[~df['A'].isin([3, 6])]

Ini mengembalikan baris yang tinggal:

   A  B
0  5  1
3  4  5

Menggunakan kaedah isin, anda boleh memilih atau mengecualikan baris dengan mudah berdasarkan senarai nilai, menyediakan penyelesaian yang lebih serba boleh untuk pengekstrakan data daripada bingkai data Pandas.

Atas ialah kandungan terperinci Bagaimana untuk Subset DataFrame Pandas Berdasarkan Senarai Nilai?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!