将多维numpy数组中的最后一个元素设置为等于第一个元素

Question

例如，我有一个数组：

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

我如何将4设置为1？我用

array[-1][-1][-1][-1] = array[0][0][0][0]

，但是后来我收到了一个错误。有更通用的方法吗？

Answer 1

您可以通过更新展平的数组来“作弊”：

a = np.array([[[1,2],[3,4]],[[1,2],[3,4]]])

a.flat[-1] = a.flat[0]

a
array([[[1, 2],
        [3, 4]],

       [[1, 2],
        [3, 1]]])