通过预定义数组分配numpy数组元素

Question

我有数组newx 和newy，大小分别为nx*ns 和ny*ns，其中nx!=ny。
我希望能够通过以下方式在数组f 中设置newx 和newy 定义的元素：

f = np.zeros([nx,ny,ns])
for s in range(ns):
    f[newx[:,s],newy[:,s],s] = s

不幸的是，这给出了一个错误：

ValueError: shape mismatch: objects cannot be broadcast to a single shape

我理解错误，但我终生无法找出正确的语法。请帮忙。

编辑：提供示例代码：

import numpy as np

newx = np.array([[0,1],
                 [1,2],
                 [2,3],
                 [3,0]])
newy = np.array([[0,1],
                 [1,2],
                 [2,0]])

f = np.zeros([4,3,2])
for s in range(2):
    f[newx[:,s],newy[:,s],s] = s

Answer 1

newx 和 newy 必须具有相同的形状，所以你必须重新塑造它们。

f = np.zeros([4,3,2])

newX = newx.reshape(8,1)
newY = newy.reshape(6,)

for s in range(2):
    f[newX , newY ,s] = s


print f