Python - Tạo Pipeline trong Pandas

Để tạo pipeline trong Pandas, chúng ta cần sử dụng phương thức pipe(). Trước tiên, nhập thư viện pandas cần thiết với bí danh:

import pandas as pd

Bây giờ, tạo DataFrame:

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

Tạo pipeline và gọi hàm tùy chỉnh upperFunc() để chuyển tên cột thành chữ hoa:

pipeline = dataFrame.pipe(upperFunc)

Dưới đây là upperFunc() để chuyển tên cột thành chữ hoa:

def upperFunc(dataframe):
    # Chuyển thành chữ hoa
    dataframe.columns = dataframe.columns.str.upper()
    return dataframe

Ví dụ

Dưới đây là mã hoàn chỉnh:

import pandas as pd
# hàm chuyển tên cột thành chữ hoa
def upperFunc(dataframe):
    # Chuyển thành chữ hoa
    dataframe.columns = dataframe.columns.str.upper()
    return dataframe
# Tạo DataFrame
dataFrame = pd.DataFrame({
    "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
    "Units": [100, 150, 110, 80, 110, 90]
})
print("DataFrame ...\n", dataFrame)
# tạo pipeline sử dụng pipe()
pipeline = dataFrame.pipe(upperFunc)
# gọi pipeline
print("\nDisplaying column names in uppercase...\n", pipeline)

Kết quả

Điều này sẽ tạo ra đầu ra sau:

DataFrame ...
    Car  Units
0   BMW    100
1  Lexus    150
2   Audi    110
3 Mustang     80
4 Bentley    110
5 Jaguar     90
Displaying column names in uppercase...
    CAR  UNITS
0   BMW    100
1  LEXUS    150
2   AUDI    110
3 MUSTANG     80
4 BENTLEY    110
5 JAGUAR     90