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

Tạo Pipeline và xóa cột khỏi DataFrame - Python Pandas

Sử dụng phương thức colDrop() của thư viện pdpipe để xóa cột khỏi DataFrame Pandas. Trước tiên, nhập các thư viện pdpipe và pandas với bí danh tương ứng:

import pdpipe as pdp
import pandas as pd

Tạo một DataFrame. Ở đây chúng ta có hai cột:

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

Để xóa cột khỏi DataFrame, sử dụng phương thức ColDrop(). Ở đây chúng ta xóa cột "Units":

resDF = pdp.ColDrop("Units").apply(dataFrame)

Ví dụ hoàn chỉnh

Mã nguồn đầy đủ:

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]
    }
)
print("DataFrame ...\n", dataFrame)
# xóa một hàng với pdp
dataFrame = pdp.ValDrop(['Jaguar'], 'Car').apply(dataFrame)
print("\n DataFrame sau khi xóa hàng...\n", dataFrame)
# xóa một cột với pdp
resDF = pdp.ColDrop("Units").apply(dataFrame)
print("\nDataFrame sau khi xóa cột...\n", resDF)

Kết quả

Chương trình sẽ tạo ra kết quả sau:

DataFrame ...
   Car  Units
0  BMW    100
1 Lexus    150
2 Audi    110
3 Mustang  80
4 Bentley  110
5 Jaguar    90
Displaying after removing a row...
   Car  Units
0  BMW    100
1 Lexus    150
2 Audi    110
3 Mustang  80
4 Bentley  110
Displaying after removing a column...
   Car
0  BMW
1 Lexus
2 Audi
3 Mustang
4 Bentley