我有以下numpy数组:
# doors = [[0,1,2],[0,1,2],....,[0,1,2]
doors = np.broadcast_to(np.array([0, 1, 2]),(max_samples, 3))
# first_choice = [ [2],[1],....,[2] ]
first_choice = np.random.randint(3, size=(max_samples, 1))
不使用任何类型的循环(for、while等),我想创建一个名为result的新numpy数组,其中包含具有抑制首选项的门。你知道吗
示例:
first_choice = [ [2], [1], [1], [0] ....]
result = [ [0,1],[0,2],[0,2],[1,2] .....]
我该怎么做?使用任何图书馆,如果接受,如果必要的话,我认为numpy可以这样做的要求,但不知道如何。你知道吗
有一种可能性:
显然,要使其起作用,
first_choice
中的每个值必须在doors
的每个对应行中出现一次(而且只能出现一次)。你知道吗相关问题 更多 >
编程相关推荐