<分区> 分区>
我想通过索引数组来提取结果
a = np.array([[10,20,30,40,50,60],
[15,25,35,45,55,65]])
b = np.array([[0,1,2,2,3,4,5,2,5,1,3,4,2,2,3,4,5,2,5,1],
[0,4,5,2,5,1,3,0,2,1,2,2,3,4,5,2,5,1,3,4]])
预期输出:
a[:,b]
[[10 20 30 30 40 50 60 30 60 20 40 50 30 30 40 50 60 30 60 20]
[15 55 65 35 65 25 45 15 35 25 35 35 45 55 65 35 65 25 45 55]]
但实际上得到了:
[[[10 20 30 30 40 50 60 30 60 20 40 50 30 30 40 50 60 30 60 20]
[10 50 60 30 60 20 40 10 30 20 30 30 40 50 60 30 60 20 40 50]]
[[15 25 35 35 45 55 65 35 65 25 45 55 35 35 45 55 65 35 65 25]
[15 55 65 35 65 25 45 15 35 25 35 35 45 55 65 35 65 25 45 55]]]
我可能明白发生了什么,但我不知道如何去做才能得到正确的输出。我希望得到帮助。谢谢
