python - 按其他列表过滤列表系列

Question

我想按列表过滤一系列列表。下面我介绍了一个MWE：

a = np.array([[0,1,2],[2,3,5]])
b = pd.Series(list(a))

所需的输出是

0 [0,1,2]

我试过了

b[b == [1, 2, 3]]
# and
b[b in [[1, 2, 3]]]

但出现错误信息('Lengths must match to compare', (2,), (3,))

标签： pythonfilter

Answer 1

谢谢你的时间！我从下面的代码解决了这个问题：

cond = [b[i]== [0, 1, 2] for i in range(b.shape[0])]
b.where(cond).dropna()
#output
0    [0, 1, 2]
dtype: object