从字典创建不同的数据框

Question

我已经创建了一本字典，方法是使用这种编码将一些雨量计按它们的代码分组

dict_of_gauges = {k: v for k, v in PE_14.groupby('gauge_code')}

这给了我一些如下所示的条目

 11800  261070705A  PAULISTA    PE 2014-08-21 17:10:00      0.2
 11801  261070705A  PAULISTA    PE 2014-08-21 17:20:00      0.0
 11802  261070705A  PAULISTA    PE 2014-08-21 17:30:00      0.2
 11803  261070705A  PAULISTA    PE 2014-08-21 17:40:00      0.0
 11804  261070705A  PAULISTA    PE 2014-08-21 18:00:00      0.0
[3966 rows x 5 columns],
 '261070704A':        gauge_code      city state            datetime  rain_mm
 11493  261070704A  PAULISTA    PE 2014-08-21 21:20:00      0.2
 11494  261070704A  PAULISTA    PE 2014-08-21 21:30:00      0.0
 11495  261070704A  PAULISTA    PE 2014-08-21 21:40:00      0.0
 11496  261070704A  PAULISTA    PE 2014-08-21 21:50:00      0.0
 11497  261070704A  PAULISTA    PE 2014-08-21 22:00:00      0.0
[4180 rows x 5 columns],

现在我真的很想为它们中的每一个创建数据框，并指定诸如“df1”、“df2”等名称……但我似乎不知道如何在 FOR 中做到这一点。 我最终使用的代码是

df1 = pd.DataFrame.from_records(dict_of_gauges['261070703A'])
df2 = pd.DataFrame.from_records(dict_of_gauges['261070705A'])
.
.
.

但是多次做同样的事情不是很专业，我不知道如何分配这些名称，我尝试制作的伪代码（如下）并没有真正起作用，因为它覆盖了“df”正如预期的那样，每个循环。

listdfs = ['df0','df1','df2','df3','df4','df5']
for df, gauge in zip(listdfs, dict_of_gauges):
    df = pd.DataFrame.from_records(dict_of_gauges[gauge]) 

有人可以给我一些启示吗？

Answer 1

可能最好的方法是只使用groupby() 的结果。

>>> gb = PE_14.groupby('gauge_code')
>>> df0 = gb.get_group("261070705A")     # Get a single group.
>>> list(gb.groups)
['261070705A', '261070704A', ...]

因此，如果您需要遍历不同的组，请执行以下操作

>>> for key in gb.groups:
...     groupdf = gb.get_group(key)
...     # ... do something with the group data frame.

在for 循环中创建重复的新变量是一个问题，但您可以很容易地制作一个数据帧字典，使它们易于跟踪，而无需每次都重新计算 groupby。事实上，这就是您在第一行代码中已经完成的工作！不要使用df1，只需使用dict_of_gauges['261070705A']。如果这太冗长并且您不关心密钥，您也可以将它们放在一个列表中：

>>> gauge_dfs = [gb.get_group(key) for key in gb.groups]

另一方面，dict_of_gauges['261070705A'] 比 gauge_dfs[0] 更具可读性。