Bagaimanakah Saya Boleh Mencipta Lajur Baharu dalam Panda Berdasarkan Nilai daripada Lajur Sedia Ada?

Buat Lajur Baharu dengan Nilai Dipilih daripada Lajur Sedia Ada

Memilih nilai tertentu daripada lajur sedia ada untuk mengisi lajur baharu ialah tugas manipulasi data biasa. Mari kita terokai dua pendekatan untuk mencapai ini dalam Python menggunakan Panda.

Kaedah 1: np.where for Two Choices

Apabila hanya terdapat dua pilihan untuk dipilih , fungsi np.where ialah pilihan yang mudah. Dalam contoh yang diberikan, kami ingin mencipta lajur warna dengan 'hijau' untuk nilai dalam lajur Tetapkan sama dengan 'Z' dan 'merah' sebaliknya.

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

Kaedah 2: np .select untuk Pelbagai Pilihan

Untuk senario dengan lebih daripada dua pilihan, np.select menawarkan lebih banyak fleksibiliti. Mari perkenalkan lebih banyak syarat untuk lajur warna:

• 'kuning' untuk bila (df['Set'] == 'Z') & (df['Type'] == 'A')
• 'biru' untuk bila (df['Set'] == 'Z') & (df['Jenis'] == 'B')
• 'ungu' untuk bila (df['Jenis'] == 'B')
• 'hitam' untuk semua kes lain

Kod untuk senario ini adalah seperti berikut:

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')

Pendekatan ini membolehkan penyesuaian mudah bagi syarat dan nilai untuk dipilih, menjadikan ia sesuai untuk senario yang lebih kompleks.

Atas ialah kandungan terperinci Bagaimanakah Saya Boleh Mencipta Lajur Baharu dalam Panda Berdasarkan Nilai daripada Lajur Sedia Ada?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!