我有两个numpy数组:
import numpy as np
arrlist = np.array([[1,0,0,1] , [0,1,1,0]])
rng = np.random.default_rng()
rand_arr = rng.choice([0, 1], size=(5, 4)))
如何找到rand_arr
中每一行的索引i
,其中rand_arr[i]
在arrlist
中?
例如,rand_arr
可能-rand_arr = np.array([[1,0,0,1],[0,1,1,0],[1,1,1,0]])
所以预期收益应该是-np.array([[0,1]])
注意,所有上述值rand_arr
和arrlist
仅在运行时已知。
1条答案
按热度按时间5uzkadbs1#
使用广播将
arrlist
的每一行与rand_arr
进行比较,然后沿着行进行聚合,最后沿where
查找有效的行索引: