python - 将 np.where 数组转换为列表

Question

我尝试使用 np.where 获取数组的索引，并希望以给我一个一维列表的方式连接列表。是否可以？

l = np.array([10,20,14,10,23,5,10,1,2,3,10,5,6,5,10])
y= np.where(l==10)
p=np.where(l==5)

如果我打印 y 和 p，他们会给我

(array([ 0,  3,  6, 10, 14]),)

(array([ 5, 11, 13]),)

追加后会生成一个元组列表。然而我想要的输出是这样的:

[0,3,6,10,14,5,11,13]

Answer 1

由于还有很多其他解决方案，我将向您展示另一种方法。

您可以使用np.isin来测试数组中的正确值:

goodvalues = {5, 10}
np.where(np.isin(l, goodvalues))[0].tolist() #  [0, 3, 6, 10, 14, 5, 11, 13]