如何编写一个改变DataFrame列顺序的函数？(Python)

import pandas as pd

def change_column_order(df, new_order):
    """
    改变DataFrame列顺序的函数

    参数:
    df (pd.DataFrame): 需要改变列顺序的DataFrame
    new_order (list): 列的新顺序列表

    返回:
    pd.DataFrame: 列顺序改变后的DataFrame
    """
    # 检查新顺序列表中的列是否都在原DataFrame中
    if not set(new_order).issubset(df.columns):
        raise ValueError("新顺序列表中包含不在原DataFrame中的列")
    
    # 创建一个新的DataFrame，只包含新顺序列表中的列
    new_df = df[new_order]
    
    return new_df

# 示例使用
if __name__ == "__main__":
    # 创建一个示例DataFrame
    data = {
        'A': [1, 2, 3],
        'B': [4, 5, 6],
        'C': [7, 8, 9]
    }
    df = pd.DataFrame(data)
    
    # 定义新的列顺序
    new_order = ['C', 'A', 'B']
    
    # 调用函数改变列顺序
    new_df = change_column_order(df, new_order)
    
    # 打印结果
    print(new_df)