numpy.array 过滤问题

dewei
  • 0
新手上路,请多包涵

请问, 比如 a = np.array([[1,2,3], [1,4,5]])
接下来我想过滤出包含 1 和 3的元素, 按照a来说就是 [1,2,3] 应该怎么过滤出来呢?
我使用了一个方法类似 a[(a[:,:]==1) & a[:,:]==3] 返回空. 求各位大牛指点迷津 谢谢

回复
阅读 216
1 个回答
import numpy as np

a = np.array([[1, 2, 3], [1, 4, 5]])
b = [i for i in a if 1 in i and 3 in i]
print(b)
你知道吗?

宣传栏