我有一个尺寸为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个值。
最佳答案
看起来像
np.array(coefs[:8]+[1]).reshape((3,3))
应该完全按照您的要求去做吧?
关于python - numpy:填充数组的子集,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/28665443/