python - 使用列表中的元素作为数据框名称

Question

我有一个这样的列表:

network = ['facebook','organic',instagram']

然后我创建了 3 个数据帧:facebook_count、organic_count、instagram_count 每种类型的网络。

facebook_count = df[df.isin(['facebook installs'])]
organic_count = df[df.isin(['organic installs'])]
instagram_count = df[df.isin(['instagram installs'])]

那么有没有一种方法可以编写一次创建这 3 个数据帧的迭代？我这样写:

for i in range(len(network)+1):
    network[i]+'_count' = df[df.isin([network[i]+' installs'])]

但它返回一个 SyntaxError: can't assign to operator

Answer 1

最有效的解决方案是创建一个存储数据帧的字典:

d = {i + '_count': df[df.isin([i + ' installs'])] for i in network}

要获取数据框，请执行以下操作:

print(d['facebook_count'])

输出将是具有预期值的 facebook_count 数据框