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

Python Pandas - Điền giá trị thiếu cột bằng trung vị (median)

Trung vị (median) tách nửa trên và nửa dưới của dữ liệu. Dùng fillna() và đặt median để điền giá trị thiếu.

Nhập thư viện:

import pandas as pd
import numpy as np

Tạo DataFrame 2 cột, đặt NaN bằng np.NaN:

dataFrame = pd.DataFrame({
    "Car": ['Lexus', 'BMW', 'Audi', 'Bentley', 'Mustang', 'Tesla'],
    "Units": [100, 150, np.NaN, 80, np.NaN, np.NaN]
})

Tìm median cột Units, thay NaN bằng median:

dataFrame.fillna(dataFrame['Units'].median(), inplace=True)

Ví dụ

import pandas as pd
import numpy as np

dataFrame = pd.DataFrame({
    "Car": ['Lexus', 'BMW', 'Audi', 'Bentley', 'Mustang', 'Tesla'],
    "Units": [100, 150, np.NaN, 80, np.NaN, np.NaN]
})
print("DataFrame ...\n", dataFrame)

dataFrame.fillna(dataFrame['Units'].median(), inplace=True)
print("\nUpdated Dataframe after filling NaN values with median...\n", dataFrame)

Đầu ra

DataFrame ...
      Car  Units
0   Lexus  100.0
1     BMW  150.0
2    Audi    NaN
3  Bentley   80.0
4  Mustang    NaN
5   Tesla    NaN

Updated Dataframe after filling NaN values with median...
      Car  Units
0   Lexus  100.0
1     BMW  150.0
2    Audi  100.0
3  Bentley   80.0
4  Mustang  100.0
5   Tesla  100.0