Computer >> Máy Tính >  >> Lập trình >> Python

Viết mã Python để chuyển đổi một chuỗi nhất định thành một biến giả và loại bỏ bất kỳ giá trị NaN nào nếu chúng tồn tại

Giả sử, bạn có một chuỗi và kết quả để chuyển đổi thành biến giả là,

   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 vấn đề này, chúng tôi 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ử 'Nam' và 'Nữ' và gán thành Chuỗi.

  • Áp dụng hàm get_dummies bên trong chuỗi và đặt giá trị dummy_na là False. Nó được định nghĩa bên dưới,

pd.get_dummies(series, dummy_na=False)

Ví dụ

Hãy kiểm tra đoạn 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