在3D数组中找到最大值的索引并存储为2D数组

Question

我有一个7x10000x10000形状的3D数组。数组的第一维是长度为7的向量。我想创建一个二维数组，形状为10000x10000，在向量中具有max元素的索引。我尝试使用循环来执行此操作，但是由于数组的大小，它非常慢。有没有更好的方法，例如展平，在每个向量中找到最大值并重塑形状？

    for i in range(0,7):
    for j in range(0,10000):
        for k in range(0,10000):
            vector = myarray[:,j,k]
            m = np.argmax(vector) 
            img[j,k] = m