命名数据框将其作为函数中的参数传递

Question

我想知道如何构建一个函数，我可以在其中传递我将创建的数据框的名称作为参数。

这里我有一个例子：

list1 = [1,2,3]
list2 = [4,5,6]

def say_hi(list1, list2):
    lists = zip(list1, list2)
    df = pd.DataFrame(lists, columns=('A', 'B'))
    return df

我创建了一个名为 df 的数据框，其中包含 3 行和 2 列。

我的目标是：

def say_hi(list1, list2, DATAFRAME_NAME):
    lists = zip(list1, list2)
    DATAFRAME_NAME = pd.DataFrame(lists, columns=('A', 'B'))
    return DATAFRAME_NAME

我创建了一个名为 DATAFRAME_NAME 的数据框，其中包含 3 行和 2 列。 我通过一个论点给它起了名字。

显然你不能以这种方式申请，但我想知道我该怎么做。

Answer 1

使用 globals() 函数代替：

import pandas as pd

def say_hi(list1, list2, df_name):
    lists = zip(list1, list2)
    df = pd.DataFrame(lists, columns=('A', 'B'))
    globals()[df_name] = df
    return df

list1 = [1,2,3]
list2 = [4,5,6]

say_hi(list1, list2, "my_df")
print(my_df)