Python创建指定对角线值的N×N矩阵报错，求解决方案

错误原因 #

• 数据类型不匹配：初始化矩阵时指定了dtype=int，但需要存储的是浮点数（如1/4、0.5这类非整数），整数类型无法容纳这些值，会导致后续赋值异常。
• range函数参数错误：range()的输入必须是整数，而1/n是浮点数（n=5时为0.2），直接传入会触发TypeError；同时这个写法也无法生成你需要的对角线数值序列。

正确代码实现 #

根据你提供的示例矩阵（5×5），可以明确需求：

• 对于n×n矩阵，行索引i（0 ≤ i < n）：
  • 对角线元素a[i,i] = i/(n-1)（对应示例中i=0→0/4=0，i=1→1/4，…，i=4→4/4=1）
  • 右侧相邻元素a[i,i+1] = 1 - i/(n-1)（仅当i < n-1时存在）

以下是可直接运行的代码：

import numpy as np

n = 5  # 矩阵维度为n×n
a = np.zeros((n, n), dtype=float)  # 使用float类型存储浮点数

# 生成对角线数值序列并填充
diag_values = np.arange(n) / (n - 1)
np.fill_diagonal(a, diag_values)

# 填充每行右侧相邻的元素
for i in range(n - 1):
    a[i, i + 1] = 1 - diag_values[i]

print(a)

运行后输出与你给出的P_4一致：

[[0.   1.   0.   0.   0.  ]
 [0.   0.25 0.75 0.   0.  ]
 [0.   0.   0.5  0.5  0.  ]
 [0.   0.   0.   0.75 0.25]
 [0.   0.   0.   0.   1.  ]]

如果你的需求是对角线元素为i/n（分母为n而非n-1），只需将代码中的n-1替换为n即可，此时最后一行对角线元素为(n-1)/n。

内容的提问来源于stack exchange，提问作者Ahmad Morwat