Viết mã Python chuyển Series cho trước thành biến giả (dummy variable) và loại bỏ giá trị NaN nếu có

Giả sử bạn có một Series và kết quả chuyển thành dummy variable như sau:

   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

Để giải quyết bài toán này, chúng ta sẽ làm theo các bước dưới đây:

Giải pháp

• Tạo danh sách với các phần tử 'Male' và 'Female' và gán vào Series
• Áp dụng hàm get_dummies trên Series và đặt dummy_na=False

pd.get_dummies(series, dummy_na=False)

Ví dụ

Hãy kiểm tra mã sau để hiểu rõ hơn:

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

Đầu ra

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