python - 创建具有相似拼写的 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()