Computer >> Máy Tính >  >> Lập trình >> Python

Tạo một đường ống và xóa một cột khỏi DataFrame - Python Pandas

Sử dụng phương thức colDrop () của thư viện pdpipe để xóa một cột khỏi Pandas DataFrame. Đầu tiên, hãy nhập các thư viện pdpipe và gấu trúc được yêu cầu với các bí danh tương ứng của chúng -

import pdpipe as pdp
import pandas as pd

Hãy để chúng tôi 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 một cột khỏi DataFrame, hãy sử dụng phương thức ColDrop (). Ở đây, chúng tôi đang xóa cột "Đơn vị" -

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

Ví dụ

Sau đây là mã hoàn chỉnh -

import pdpipe as pdp
import pandas as pd

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

print("DataFrame ...\n",dataFrame)

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

# removing a column with pdp
resDF = pdp.ColDrop("Units").apply(dataFrame)
print("\nDataFrame after removing a column...\n",resDF)

Đầu ra

Điều này 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