python - 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))

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