我的数据框具有列表作为元素,我希望有一种更有效的方法来检查某些条件。
我的数据框看起来像这样
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/