python - 在循环中创建多个数据框

Question

我有一个人口数据。我想为每个州和年份创建单独的数据框。思路如下:

for i in province_id:
    for j in year:
         sub_data_i_j = data[(data.provid==i) &(data.wave==j)]

但是，我不确定如何动态生成 sub_data_i_j。

Answer 1

应该这样做:

for i in province_id:
    for j in year:
        locals()['sub_data_{}_{}'.format(i,j)] = data[(data.provid==i) & (data.wave==j)]

我最初建议使用 exec，出于安全原因，这通常不被认为是最佳实践。话虽如此，如果您的代码没有暴露给任何有恶意的人，那应该没问题，为了完整起见，我将其保留在这里:

for i in province_id:
    for j in year:
        exec "sub_data_{}_{} = data[(data.provid==i) & (data.wave==j)]".format(i,j)

然而，对于大多数用例，最好使用某种集合，例如一个字典，因为在代码的后续部分引用动态生成的变量名会很麻烦。它也是单线的:

data_dict = {key:g for key,g in data.groupby(['provid','wave'])}