如何构建Python函数来动态构建Pandas DataFrame？

Question

我在熊猫中有一个数据框，需要用它来创建其他数据框。

数据框包含naic代码以及相关数据。实际上，我正在尝试为每个代码创建一个新的数据框，并陷入错误。

fdf是一个具有2位数字的数据帧，即：10,11,12,13。

我想遍历此数据框以查询和构建其他许多数据。这是我到目前为止的内容：

for x in fdf:
    'Sdf' + str(x) = df[df['naics'].astype(str).str[2:4]==str(x)]

如果我自己运行它：

df[df['naics'].astype(str).str[2:4]==str(57)]

它返回我想要的数据帧，但是我不确定如何将其构建为函数。

“ SyntaxError：无法分配给函数调用”是我得到的错误。我认为问题是我如何尝试动态构建数据框名称？

非常感谢您的帮助。

Answer 1

使用字典来做。

df_list = {}
for x in fdf:
    df_list[str(x)] = df[df['naics'].astype(str).str[2:4]==str(x)]