python - 基于列表对 Python 3.x 数据框进行子集化

Question

我有一个数据框df和一个包含多个用户ID的列表in_list。我想根据 in_list 的用户 ID 创建多个数据帧。

这是列表:

in_list = 
[4638472273,
 1559410755,
 4931532174,
 2419930464,
 1884182865,
 3688089071,
 4555003213,
 2068627935,
 2894365987,
 8549533077]

这是代码:

i = 0
while i < len(in_list):
    user_index[i] = df[(df.In == in_list[i])] 
    i += 1

当我执行上述代码时，我收到此错误消息ValueError:无法设置没有定义索引的框架和无法转换为系列的值。

我想使用数据帧 df< 的子集创建多个名为 user_index0、user_index1、user_index2 等的数据帧但我收到此错误。

Answer 1

您可以使用isin执行类似的操作来过滤数据帧，以首先匹配“in_list”，然后groupby“in”和append结果将组添加到“user_index”列表:

user_index = []
for _,g in df[df['in'].isin(in_list)].groupby('in'):
    user_index.append(g)