我有下面的numpy数组
numpy
import numpy as np np.identity(13)
现在我想用其他数字,比如0.45,来替换所有非对角线元素。是否有任何直接的方法可用于执行此操作?
0g0grzrc1#
那么下面呢?
import numpy as np n = 13 val_offdiag = 0.45 val_diag = 1 a = np.full((n ,n), val_offdiag) - np.identity(n) * (val_offdiag - val_diag)
e4yzc0pl2#
您可以使用numpy.where
numpy.where
np.where(np.identity(13)==0, 0.45, 1)
2条答案
按热度按时间0g0grzrc1#
那么下面呢?
e4yzc0pl2#
您可以使用
numpy.where