Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

เขียนโค้ด Python แปลง Series ที่กำหนดเป็น Dummy Variable และ Drop ค่า NaN หากมี


สมมติ คุณมี Series และผลลัพธ์สำหรับการแปลงเป็น Dummy Variable คือ

   Female  Male
0       0     1
1       1     0
2       0     1
3       1     0
4       0     1
5       0     0
6       1     0
7       1     0

เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนด้านล่าง −

Solution

  • สร้าง List ที่มี 'Male' และ 'Female' และ Assign เข้า Series

  • ใช้ฟังก์ชัน get_dummies ภายใน Series และตั้งค่า dummy_na เป็น False คือ ด้านล่าง

pd.get_dummies(series, dummy_na=False)

ตัวอย่าง

import pandas as pd
import numpy as np
gender = ['Male','Female','Male','Female','Male',np.nan,'Female','Female',]
series = pd.Series(gender)
print("Series is:\n",series)
print("Dummy code is:\n", pd.get_dummies(series, dummy_na=False))

ผลลัพธ์

Series is:
0     Male
1     Female
2     Male
3     Female
4     Male
5     NaN
6     Female
7     Female
dtype: object
Dummy code is:
   Female  Male
0       0     1
1       1     0
2       0     1
3       1     0
4       0     1
5       0     0
6       1     0
7       1     0