numpy：填充数组的子集

Question

我有一个3x3维度的numpy数组，即它总共有9个元素。我有8个值，我想设置，位置（2,2）的值总是1.我可以单独完成，如下所示：

import numpy as np
def set(coefs):
    a = np.zeros(3,3)
    a[0, 0] = coefs[0]
    a[0, 1] = coefs[1]
    a[0, 2] = coefs[2]
    a[1, 0] = coefs[3]

    ...
    a[2, 2] = 1

但是，我想知道是否有更简洁和通用的方法来填补这8个值。

Answer 1

似乎

np.array(coefs[:8]+[1]).reshape((3,3))

应该完全符合您的要求，对吧？