Skip to main content
 首页 » 编程设计

python之通过二维索引提取轴

2025年05月04日23wuhuacong

<分区>

我想通过索引数组来提取结果

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]]] 

我可能明白发生了什么,但我不知道如何去做才能得到正确的输出。我希望得到帮助。谢谢