Python - สร้าง Pipeline และลบแถวจาก DataFrame ที่สร้างแล้วด้วย pdpipe

บทนำ

ใช้ไลบรารี pdpipe สร้าง Pipeline ลบแถวจาก DataFrame ด้วย ValDrop()

ขั้นตอน

1. นำเข้า pdpipe และ pandas
2. สร้าง DataFrame ข้อมูลรถยนต์
3. เพิ่มคอลัมน์ Stock จากเงื่อนไข Units
4. ลบแถวที่ Car == 'Jaguar' ด้วย pdp.ValDrop(['Jaguar'], 'Car').apply(dataFrame)

โค้ดเต็ม

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 with a new column...\n", dataFrame)

dataFrame = pdp.ValDrop(['Jaguar'], 'Car').apply(dataFrame)
print("\nDataFrame after removing a row...\n", dataFrame)

ผลลัพธ์

แถว Jaguar ถูกลบออก เหลือ 5 แถว