命名數據框將其作為函數中的參數傳遞

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)