I have a dataframe df like this:
col1 | col2
a | [1,2]
b | [3,4]
c | [3,9]
我想根据匹配的输入数组获取行,因此如果我有数组[1,2],我可以得到:
col1 | col2
a | [1,2]
当我尝试使用这个公式时,它不起作用:
df.loc[df['Col2'] == [1,2]]
Error: Lengths must match to compare
df[[v == [1, 2] for v in df['col2'].tolist()]]
df.apply
,但它比这个方法慢得多。