+-
我看到了一个函数numpy.fill_diagonal,它为对角线元素分配了相同的值.但是我想为每个对角线元素分配不同的随机值.我如何在 python中做到这一点？可能正在使用scipy或其他库？
最佳答案
您可以使用
np.diag_indices获取这些索引,然后使用这些索引简单地索引到数组中并分配值.
这是一个示例运行来说明它-
In [86]: arr # Input array
Out[86]:
array([[13, 69, 35, 98, 16],
[93, 42, 72, 51, 65],
[51, 33, 96, 43, 53],
[15, 26, 16, 17, 52],
[31, 54, 29, 95, 80]])
# Get row, col indices
In [87]: row,col = np.diag_indices(arr.shape[0])
# Assign values, let's say from an array to illustrate
In [88]: arr[row,col] = np.array([100,200,300,400,500])
In [89]: arr
Out[89]:
array([[100, 69, 35, 98, 16],
[ 93, 200, 72, 51, 65],
[ 51, 33, 300, 43, 53],
[ 15, 26, 16, 400, 52],
[ 31, 54, 29, 95, 500]])
您也可以使用np.diag_indices_from,并且可能会更像偶像,
row, col = np.diag_indices_from(arr)
点击查看更多相关文章
转载注明原文：为每个python numpy用不同的值填充矩阵对角线 - 乐贴网
