Bagaimana untuk Mencipta Lajur Baharu dengan Nilai Berasaskan Syarat dalam Panda?

Membuat Lajur Baharu dengan Nilai Berasaskan Keadaan

Soalan ini meneroka cara menambah lajur baharu, warna, pada bingkai data yang diberikan. Syaratnya ialah warna harus ditetapkan kepada 'hijau' jika nilai yang sepadan dalam lajur Set ialah 'Z' dan 'merah' sebaliknya.

Penyelesaian dengan Numpy Where:

Untuk senario dengan dua pilihan sahaja, kaedah np.where boleh digunakan. Berikut ialah kodnya:

df['color'] = np.where(df['Set'] == 'Z', 'green', 'red')

Pendekatan ini secara berkesan memperuntukkan 'hijau' kepada baris dengan Set ialah 'Z' dan 'merah' sebaliknya.

Penyelesaian dengan Numpy Select:

Dalam kes di mana terdapat lebih daripada dua syarat, np.select boleh digunakan. Katakan warna harus memenuhi kriteria berikut:

• 'kuning' apabila Set ialah 'Z' dan Jenis ialah 'A'
• 'biru' apabila Set ialah 'Z' dan Taip ialah 'B'
• 'ungu' apabila Jenis ialah 'B'
• 'hitam' jika tidak

Dalam senario ini, kodnya ialah:

conditions = [
    (df['Set'] == 'Z') & (df['Type'] == 'A'),
    (df['Set'] == 'Z') & (df['Type'] == 'B'),
    (df['Type'] == 'B')]
choices = ['yellow', 'blue', 'purple']
df['color'] = np.select(conditions, choices, default='black')

Penyelesaian ini membenarkan penetapan nilai berasaskan keadaan yang fleksibel dan berbutir untuk lajur baharu.

Atas ialah kandungan terperinci Bagaimana untuk Mencipta Lajur Baharu dengan Nilai Berasaskan Syarat dalam Panda?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!