在函數中創建 Pandas 數據框

Question

對於腳本，我需要在腳本開頭創建字典、列表和數據框。 為此，我這樣做：

df4 = pd.DataFrame()
object_dict = {}
caption = []

因為有很多不同的數據框，我想把它放到一個函數中，比如：

def create_data_objects():
  df4 = pd.DataFrame()
  object_dict = {}
  caption = []

但是如果我這樣做，我會得到“NameError: name 'df4' is not defined”，所以沒有創建數據框。 有沒有辦法把它放到一個函數中並使我的代碼更精簡？

Answer 1

當您在函數內創建對象時，它們的作用域會在函數結束時結束。

你可以考慮這樣做：

def create_data_objects():
  df4 = pd.DataFrame()
  object_dict = {}
  caption = []
  return df4, object_dict, caption

def MAIN_FUNCTION():
  df4, object_dict, caption = create_data_objects()