Mage 筆記：Data Integration pipeline 轉換欄位名稱

紀錄一下如何在 Data Integration pipeline 中，使用 Transform 轉換欄位名稱。

例如這是我原始 Table

UU01	UU02	UU03
51982064-26f7-435c-a935-bf522ebda29d	王曉明	1
91d49f51-495d-4d98-833e-a1e090475195	張阿榮	0
251e5455-c061-4836-b2d7-ee40b7da4f54	陳曉嵐	1
0295db65-b2e3-40ac-b2ec-24798703df1b	陳大華	2

而我想轉成：

id	name	status
51982064-26f7-435c-a935-bf522ebda29d	王曉明	1
91d49f51-495d-4d98-833e-a1e090475195	張阿榮	0
251e5455-c061-4836-b2d7-ee40b7da4f54	陳曉嵐	1
0295db65-b2e3-40ac-b2ec-24798703df1b	陳大華	2

Transform

要進行以下修改，要使用 transform

code 如下

from mage_ai.data_cleaner.transformer_actions.base import BaseAction
from mage_ai.data_cleaner.transformer_actions.constants import ActionType, Axis
from mage_ai.data_cleaner.transformer_actions.utils import build_transformer_action
from pandas import DataFrame

if 'transformer' not in globals():
    from mage_ai.data_preparation.decorators import transformer
if 'test' not in globals():
    from mage_ai.data_preparation.decorators import test


@transformer
def transform(df: DataFrame, *args, **kwargs) -> DataFrame: 
    """
    欄位轉換，這段程式會根據 mapping 變數的 key / value 轉換匯出的欄位
    """

    mapping = {
        "UU01": "id",
        "UU02": "name",
        "UU03": "status"
    }
    
    # 實際執行 rename；不存在的欄位會被忽略
    df_renamed = df.rename(columns=mapping)

    return df_renamed



@test
def test_output(output, *args) -> None:
    """
    Template code for testing the output of the block.
    """
    assert output is not None, 'The output is undefined'

執行完畢後，就會匯出欄位名稱是 id / name / status