如何将CSV文件作为输入参数传入自定义函数

你当前传入function('sample.csv')的只是代表文件路径的字符串，函数本身没有内置读取本地文件、解析CSV格式的逻辑，自然只会将输入识别为普通字符串，无法完成后续CSV处理。

根据你的使用场景，可以选择以下两种改造方式：

方案1：保留按路径传入的调用形式 #

如果你希望继续使用function('sample.csv')的调用方式，直接在函数内部新增文件读取、CSV解析的逻辑即可。以下是Python环境的示例代码：

import csv

def function(csv_path: str):
    # 校验并读取本地CSV文件
    try:
        with open(csv_path, 'r', encoding='utf-8') as file:
            # 可根据需求选择csv.reader（返回列表）或csv.DictReader（返回字典）
            csv_data = [row for row in csv.DictReader(file)]
    except FileNotFoundError:
        raise ValueError("传入的CSV文件路径不存在")
    
    # 此处添加你原本的CSV处理逻辑，直接操作csv_data即可
    return csv_data

• 适用场景：调用方仅需要传入本地文件路径，不需要额外处理文件读取逻辑。

方案2：函数调整为接收解析后的CSV内容 #

如果你希望函数兼容性更强，不绑定本地文件读取逻辑，可以调整函数的入参要求，调用前先完成CSV读取解析再传入：

import csv

# 调用侧先完成CSV读取解析
with open('sample.csv', 'r', encoding='utf-8') as file:
    parsed_csv = [row for row in csv.DictReader(file)]

# 传入解析后的内容
function(parsed_csv)

• 适用场景：需要兼容多数据源的情况，比如内存生成的CSV数据、接口返回的CSV内容，不需要必须落地为本地文件。

如果你传入的是CSV格式的纯文本字符串而非文件路径，只需要在函数内部新增CSV字符串解析逻辑即可，不需要走本地文件读取步骤，不同语言都有对应的CSV字符串解析工具可以直接调用。

内容的提问来源于stack exchange，提问作者jt25