如何使用花式索引和使用2D索引数组的尺寸切片numpy 3D数组

时间:2020-08-13 03:04:28

标签: python-3.x numpy-ndarray numpy-slicing

因此,我正在尝试刷新我的知识。我尝试找到此问题here的答案,但仍不确定如何回答。

基本上,如果我有一个形状为(x,z)的2D索引数组(称为a)和一个形状为(x,y,z)的3D数组(称为b),那么我想要一个2D数组(称为c)具有以下属性:c的i,j元素c [i,j] = b [i,a [i,j],j]。 c的形状是(x,z)

     {
        "detail": "No active account found with the given credentials"
     }

我尝试使用以下代码,但没有获得我想要的结果。

To summarize:
a.shape = (x, z)
b.shape = (x, y, z)   
c[i, j] = b[i, a[i, j], j]
c.shape = (x, z)

结果是

import numpy as np

input = numpy.random.randn(3, 5, 10)
target_np = np.random.randint(high = 5, low = 1, size = (3, 10))
q = input[:, np.ravel(target_np),:]

print(q.shape)

0 个答案:

没有答案