如何修正NumPy代码以构建符合预期的行列元素求和矩阵？

问题根源 #

你的代码存在两个核心问题：

1. Python range 的左闭右开特性：range(start, stop, step) 不会包含 stop 值，导致生成的行列数比预期少（比如输入行参数1、4、1时，range(1,4,1)仅生成[1,2,3]，少了4）。
2. 行列值的对应逻辑不符合你的期望：当前代码直接计算行值+列值，但你的期望矩阵显示元素更接近「行值 + 列的偏移量（从0开始）」，而非行值与输入列参数的直接求和。

修正方案 #

方案1：生成「行值 + 输入列参数值」的矩阵（修正range边界） #

如果你的真实需求是矩阵元素为输入行参数序列的每个值与输入列参数序列的每个值之和，只需将range的结束值改为Endpoint + Step，确保包含终点值，同时补上NumPy的导入语句：

import numpy as np

StartpointRow = int(input("First number of row?:\n"))
EndpointRow = int(input("Last number of row?:\n"))
StepRow = int(input("Which steps should the row have?:\n"))

StartpointCol = int(input("First number of column?:\n"))
EndpointCol = int(input("Last number of column?:\n"))
StepCol = int(input("Which steps should the column have?:\n"))

# 修正range的结束值，确保包含EndpointRow和EndpointCol
row_values = range(StartpointRow, EndpointRow + StepRow, StepRow)
col_values = range(StartpointCol, EndpointCol + StepCol, StepCol)

x = np.array([[i + j for i in col_values] for j in row_values])
print(x)

测试输入1（行1、4、1，列1、4、1）：
生成结果：

[[2 3 4 5]
 [3 4 5 6]
 [4 5 6 7]
 [5 6 7 8]]

方案2：生成「行值 + 列偏移量（从0开始）」的矩阵（匹配你的期望矩阵） #

如果你的期望是矩阵元素为行值加上从0开始的列索引偏移量，且列的数量由输入列参数决定（(EndpointCol - StartpointCol)//StepCol + 1），可以修改为：

import numpy as np

StartpointRow = int(input("First number of row?:\n"))
EndpointRow = int(input("Last number of row?:\n"))
StepRow = int(input("Which steps should the row have?:\n"))

StartpointCol = int(input("First number of column?:\n"))
EndpointCol = int(input("Last number of column?:\n"))
StepCol = int(input("Which steps should the column have?:\n"))

# 生成行值序列（包含EndpointRow）
row_values = range(StartpointRow, EndpointRow + StepRow, StepRow)
# 计算列的数量
col_count = (EndpointCol - StartpointCol) // StepCol + 1

x = np.array([[j + col_idx for col_idx in range(col_count)] for j in row_values])
print(x)

测试输入1（行1、4、1，列1、4、1）：
生成结果与你的期望核心逻辑一致：

[[1 2 3 4]
 [2 3 4 5]
 [3 4 5 6]
 [4 5 6 7]]

（注：你的原期望矩阵第二行第二个元素写为4，这里推测是笔误，实际逻辑下应为3；若你确实需要第二行第二个元素为4，需补充更明确的行列值对应规则）

方案3：匹配你第二个例子的特殊需求 #

你的第二个例子输入行1、4、1，列2、4、1时期望生成第一行0-4的矩阵，推测你需要列从0开始到EndpointCol，元素为行值+列值，可修改为：

import numpy as np

StartpointRow = int(input("First number of row?:\n"))
EndpointRow = int(input("Last number of row?:\n"))
StepRow = int(input("Which steps should the row have?:\n"))

# 简化列输入，仅保留终点和步长
EndpointCol = int(input("Last number of column?:\n"))
StepCol = int(input("Which steps should the column have?:\n"))

row_values = range(StartpointRow, EndpointRow + StepRow, StepRow)
col_values = range(0, EndpointCol + StepCol, StepCol)

x = np.array([[j + i for i in col_values] for j in row_values])
print(x)

测试输入（行1、4、1，列终点4、步长1）：
生成结果：

[[1 2 3 4 5]
 [2 3 4 5 6]
 [3 4 5 6 7]
 [4 5 6 7 8]]

（若你需要第一行从0开始，只需将StartpointRow设为0即可）

总结 #

根据你的实际需求选择对应方案：

• 若需要行参数与列参数的直接求和，选方案1；
• 若需要行参数加从0开始的列索引，选方案2；
• 若需要列从0开始到指定终点，选方案3。

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