Python 使用另一个多维数组作为索引的numpy集值

Python 使用另一个多维数组作为索引的numpy集值,python,numpy,Python,Numpy,假设有一个四维数组idx1,为另一个五维数组zeros1存储第五维索引。 比如: 如何使用numpy高级索引进行此操作。使用开放范围数组和索引进行分配- out = np.zeros( (N,T,H,W, 256) ) i,j,k,l = np.ogrid[:N,:T,:H,:W] out[i,j,k,l,idx1] = 1 或者,在一行中- out[tuple((np.ogrid[:N,:T,:H,:W]+[idx1]))] = 1 out = np.zeros( (N,T,H,W, 25

假设有一个四维数组idx1,为另一个五维数组zeros1存储第五维索引。 比如:


如何使用numpy高级索引进行此操作。

使用开放范围数组和索引进行分配-

out = np.zeros( (N,T,H,W, 256) )
i,j,k,l = np.ogrid[:N,:T,:H,:W]
out[i,j,k,l,idx1] = 1
或者,在一行中-

out[tuple((np.ogrid[:N,:T,:H,:W]+[idx1]))] = 1
out = np.zeros( (N,T,H,W, 256) )
i,j,k,l = np.ogrid[:N,:T,:H,:W]
out[i,j,k,l,idx1] = 1
out[tuple((np.ogrid[:N,:T,:H,:W]+[idx1]))] = 1