创建一个拼写相似的 pandas 数据框变量名称列表

Question

在我的环境中，我列出了几个名称相似的 pandas 数据框。

例如：

  import pandas as pd
  import numpy as np 
 df_abc = pd.DataFrame(np.random.randn(6,4), index=dates, columns=list('ABCD')
 df_xyz = pd.DataFrame(np.random.randn(6,4), index=dates, columns=list('ABCD')
 df_2017 = pd.DataFrame(np.random.randn(6,4), index=dates, columns=list('ABCD')
 ...  potentially others

我想创建一个列表，该列表会自动确定我的环境中存在哪些数据框，并将它们动态拉入列表中。

 list_of_dfs = ['df_abc','df_xyz','df_2017', 'df_anything else']  
  # except done dynamically.   In this example anything beginning with 'df_' 
#  list_of_dfs = function_help(begins with 'df_')

Answer 1

globals() 应该为全局变量返回一个 variable_name:variable_value 的字典。

如果您想要名称以“df_”开头的已定义变量列表，您可以这样做：

list_of_dfs = [variable for variable in globals().keys()
               if variable.startswith('df_')]

我认为必须有一种更好的方法，而不是全局存储数据帧，并依赖 globals() 来获取它们的变量名。也许将它们全部存储在字典中？：

dataframes = {}
dataframes['df_1'] = pd.DataFrame()
dataframes['df_2'] = pd.DataFrame()

list_of_dfs = dataframes.keys()