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

Tạo Pipeline và xóa hàng từ DataFrame đã có - Python Pandas


Dùng ValDrop() của thư viện pdpipe để xóa hàng từ DataFrame hiện có.

Nhập thư viện:

import pdpipe as pdp 
import pandas as pd

Tạo DataFrame:

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

Xóa hàng:

dataFrame = pdp.ValDrop(['Jaguar'], 'Car').apply(dataFrame)

Ví dụ

import pdpipe as pdp 
import pandas as pd 
def demo(x): 
    if x >= 100: 
        return "OverStock" 
    else: 
        return "UnderStock" 
dataFrame = pd.DataFrame({"Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
    "Units": [100, 150, 110, 80, 110, 90]}) 
print("DataFrame ...\n", dataFrame) 
dataFrame['Stock'] = dataFrame['Units'].apply(demo) 
print("\nDataFrame với cột mới...\n", dataFrame) 
dataFrame = pdp.ValDrop(['Jaguar'], 'Car').apply(dataFrame) 
print("\nDataFrame sau khi xóa hàng...\n", dataFrame)

Kết quả

DataFrame ... 
  Car  Units 
0 BMW  100 
1 Lexus  150 
2 Audi  110 
3 Mustang  80 
4 Bentley  110 
5 Jaguar  90 
DataFrame with a new column... 
  Car  Units  Stock 
0 BMW  100  OverStock 
1 Lexus  150  OverStock 
2 Audi  110  OverStock 
3 Mustang  80  UnderStock 
4 Bentley  110  OverStock 
5 Jaguar  90  UnderStock 
DataFrame after removing a value... 
  Car  Units  Stock 
0 BMW  100  OverStock 
1 Lexus  150  OverStock 
2 Audi  110  OverStock 
3 Mustang  80  UnderStock 
4 Bentley  110  OverStock