使用np.fill_diagonal填充矩阵对角线为0时打印结果为None的解决方法

np.fill_diagonal是NumPy设计的原地操作函数，执行逻辑是直接修改传入的原始矩阵，本身没有定义返回值，默认返回None。你打印出None和矩阵尺寸是1862*1862没有关系，不管传入的数组是什么形状，只要调用这个函数后直接打印它的执行结果，都会得到None，属于对函数返回值特性不熟悉导致的用法错误。

根据你是否需要保留原始矩阵，选对应写法即可：

• 不需要保留原矩阵，直接修改原数组：不要接收函数返回值，先执行填充操作，再直接打印原数组

# 假设x是你已有的1862*1862矩阵
np.fill_diagonal(x, 0)
print(x)

错误写法参考：print(np.fill_diagonal(x, 0)) 这种写法打印的是函数默认返回的None，看不到修改后的矩阵。

• 需要保留原矩阵，生成对角线为0的新矩阵：先对原矩阵做拷贝，再在拷贝后的数组上执行填充操作

x_new = x.copy()
np.fill_diagonal(x_new, 0)
print(x_new)
# 此时原矩阵x不会被修改，x_new是对角线填充为0的新矩阵

内容的提问来源于stack exchange，提问作者Segmentation fault