我的数据框具有列表作为元素,我希望有一种更有效的方法来检查某些条件。
我的数据框看起来像这样

col_a   col_b
0   100 [1, 2, 3]
1   200 [2, 1]
2   300 [3]


我只想获取col_b中具有1的那些行。
我尝试过幼稚的方式
temp_list =列表()

for i in range(len(df1.index)):
    if 1 in df1.iloc[i,1]:
        temp_list.append(df1.iloc[i,0])


对于像这样的大数据帧,这会花费很多时间。我如何使这样的数据框搜索更有效?

最佳答案

df[df.col_b.apply(lambda x: 1 in x)]


结果是:

col_a   col_b
0   100 [1, 2, 3]
1   200 [2, 1]

关于python - 用元素作为列表切片数据框,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/56129426/

10-12 16:38
查看更多