我有一个尺寸为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/

10-12 23:05