如何在函数中创建名称取自参数的DataFrame？

嘿，我懂你想要实现的需求——根据传入函数的参数值，动态创建对应名称的DataFrame对吧？其实直接在函数里生成动态命名的局部变量并不是最佳实践，这会让代码的可读性和维护性大打折扣。不过我可以给你两种可行的方案，你可以根据自己的使用场景来选：

推荐方案：返回字典存储动态命名的DataFrame #

这种方式既清晰又安全，还方便后续管理多个DataFrame，是业界常用的做法：

import pandas as pd

def Upload(file, df_suffix):
    # 根据文件类型读取数据，这里以csv为例，可替换为read_excel等
    df = pd.read_csv(file)
    # 返回字典，键是你想要的DataFrame名称前缀+参数值，值是生成的DataFrame
    return {f"df_{df_suffix}": df}

使用示例 #

调用函数后，你可以从返回的字典里提取对应的DataFrame：

# 调用函数，传入文件名和后缀"ABC"
result = Upload("your_data.csv", "ABC")
# 提取名为df_ABC的DataFrame
df_ABC = result["df_ABC"]

# 如果传入后缀"NYK"
result = Upload("your_data.csv", "NYK")
df_NYK = result["df_NYK"]

不推荐但可实现的方案：动态修改全局命名空间 #

如果你一定要在全局环境中直接生成对应名称的变量，可以用globals()函数，但这种方法会污染全局命名空间，还可能因为参数值不合法导致报错，所以只建议在特殊场景下使用：

import pandas as pd

def Upload(file, df_suffix):
    df = pd.read_csv(file)
    # 直接在全局命名空间创建名为df_xxx的变量
    globals()[f"df_{df_suffix}"] = df
    return df

使用示例 #

调用后全局环境会直接出现对应的变量：

Upload("your_data.csv", "ABC")
# 此时可以直接使用df_ABC这个变量
print(df_ABC.head())

总的来说，优先选择字典的方案，它能让你的代码更规范，也更容易调试和扩展。

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