Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Python - Chuyển đổi DataFrame Pandas sang dữ liệu nhị phân (Binary) với get_dummies


Sử dụng pd.get_dummies() trên một cột phân loại để tạo ra các cột nhị phân (one-hot encoding) cho từng giá trị duy nhất.

Ví dụ

Minh họa:

import pandas as pd
dataFrame = pd.DataFrame({
    "Student": ['Jack', 'Robin', 'Ted', 'Scarlett', 'Kat'],
    "Result": ['Pass', 'Fail', 'Fail', 'Pass', 'Pass']
})
print("DataFrame ...\n", dataFrame)
dfBinary = pd.get_dummies(dataFrame["Result"])
print("\nDataFrame dạng nhị phân (One-hot encoding):\n", dfBinary)

Kết quả

DataFrame ...
  Result Student
0   Pass    Jack
1   Fail   Robin
2   Fail     Ted
3   Pass Scarlett
4   Pass     Kat
Displaying DataFrame in Binary form...
   Fail  Pass
0     0     1
1     1     0
2     1     0
3     0     1
4     0     1