如何将argparse值传递给函数，并将值赋值给dataframe？

import argparse
import pandas as pd

# 创建命令行参数解析器
parser = argparse.ArgumentParser(description='Process some arguments.')
parser.add_argument('--input', type=str, help='Input file path')
parser.add_argument('--output', type=str, help='Output file path')
args = parser.parse_args()

# 定义一个函数，接收参数并处理
def process_data(input_path, output_path):
    # 读取输入文件为DataFrame
    df = pd.read_csv(input_path)
    
    # 在这里可以对DataFrame进行各种操作和处理
    
    # 将处理后的结果保存到输出文件
    df.to_csv(output_path, index=False)

# 调用函数，并传递argparse的参数值
process_data(args.input, args.output)