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

เขียนโค้ด Python เพื่อแปลงชุดข้อมูลให้เป็นตัวแปรจำลอง และวางค่า NaN หากมีอยู่


สมมติว่าคุณมีชุดข้อมูลและผลลัพธ์สำหรับการแปลงเป็นตัวแปรจำลองเป็น

   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

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

วิธีแก้ปัญหา

  • สร้างรายการที่มีองค์ประกอบ "ชาย" และ "หญิง" และกำหนดลงในซีรีส์

  • ใช้ฟังก์ชัน get_dummies ภายในชุดข้อมูลและตั้งค่า dummy_na เป็นเท็จ มีการกำหนดไว้ด้านล่าง

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