如何用数字替换ndarray？

Question

我有以下代码：

arr = zip(*people2)
for i in range(len(arr)):
    for j in range(len(arr[i])):
        for k in range(len(arr[i][j])):
            if(arr[i][j][k] == 1):
                arr[i][j] = k
                break

有一组数组数组。我需要用数字替换这些最后的数组。 像这样：

[
 [[1, 0, 0], [2, 0, 0]],
 [[3, 0, 0], [4, 0, 0]],
]

->

[
 [1, 2],
 [3, 4]
]

这里的数字只是例如。

我该怎么做？我尝试使用numpy.resize()，但它不起作用。