因此,我正在尝试刷新我的知识。我尝试找到此问题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)