我有一个数组A，其形状为[2，512，21128]，另一个列表为[7，41]我想从A[0，7，：]和A[1，41，：]创建一个新的数组，其形状为[2，21128]。我可以用循环来做这样的事情，有没有什么方便的方法来做呢？我试着在网上找到它，但我实际上不知道如何问这个问题。
ve7v8dk21#
您可以使用高级索引：
arr = np.random.rand(2, 512, 21128) out = arr[[0, 1], [7, 41]]
out.shape：
out.shape
(2, 21128)
1条答案
按热度按时间ve7v8dk21#
您可以使用高级索引：
out.shape：