如何遍历CSV文件中lat、lon、id行值作为参数调用audience函数

实现方案 #

你可以直接用pandas读取CSV文件后逐行遍历调用函数即可，示例代码如下：

import pandas as pd
import os

# 提前创建输出目录避免导出时报错
os.makedirs("output", exist_ok=True)

# 替换为你实际的CSV文件路径
param_df = pd.read_csv("你的参数文件路径.csv")

# 逐行读取参数调用函数
for _, row in param_df.iterrows():
    audience(
        lat=row["lat"],
        lon=row["lon"],
        ids=row["id"]
    )

如果需要追踪执行进度、避免单行报错导致整体程序中断，可以加上日志和异常处理：

import pandas as pd
import os

os.makedirs("output", exist_ok=True)
param_df = pd.read_csv("你的参数文件路径.csv")
total_rows = len(param_df)

for idx, row in param_df.iterrows():
    current_id = row["id"]
    print(f"正在处理 {idx+1}/{total_rows}，当前ID：{current_id}")
    try:
        audience(row["lat"], row["lon"], current_id)
        print(f"ID {current_id} 处理完成")
    except Exception as e:
        print(f"ID {current_id} 处理失败，错误信息：{str(e)}")

注意事项 #

• 请将代码中的你的参数文件路径.csv替换为你本地实际的CSV文件路径
• 确保你的CSV文件中三个字段的列名和代码中读取的lat、lon、id完全一致，如果你的列名有差异，修改代码中括号内的列名即可
• os.makedirs("output", exist_ok=True)会自动创建不存在的output目录，避免函数导出CSV时报找不到路径的错误

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