如何修复无循环回代法解上三角矩阵时的TypeError：仅整数标量数组可转标量索引？

错误原因分析 #

你遇到的TypeError是因为索引方式不符合NumPy的规则：当i和j都是数组时，R[i, j::-1]会被解析为花式索引（逐元素配对取单个元素），而不是你期望的“每行取从j到末尾的切片”。比如当i=[1,0]、j=[2,1]时，j::-1是[1,2]，NumPy会取R[1,1]和R[0,2]这两个孤立元素，而非R[1,2:]和R[0,1:]这样的行切片，从而触发索引错误。

修复后的无循环代码 #

以下是符合要求的无循环实现，利用NumPy的向量化操作和einsum计算逐行点积：

import numpy as np

def backward(R, c):
    n = len(c)
    x = np.zeros(n)
    # 先计算最后一个元素（回代法的起始点）
    x[-1] = c[-1] / R[-1, -1]
    
    # 生成需要从后往前计算的元素索引
    i = np.arange(n-2, -1, -1)
    # 生成每个索引对应的R行切片和x切片
    r_slices = np.array([R[k, k+1:] for k in i])
    x_slices = np.array([x[k+1:] for k in i])
    
    # 计算每行的点积（sum(R[i,j] * x[j] for j > i)）
    dot_products = np.einsum('ij,ij->i', r_slices, x_slices)
    # 代入回代公式计算x[i]
    x[i] = (c[i] - dot_products) / R[i, i]
    
    return x

代码说明 #

1. 起始元素计算：回代法从最后一个元素开始，直接用x[-1] = c[-1] / R[-1, -1]计算。
2. 索引生成：i = np.arange(n-2, -1, -1)生成从倒数第二个元素到第一个元素的索引数组，对应需要从后往前计算的位置。
3. 切片生成：通过列表推导式生成每个索引k对应的R行切片（R[k, k+1:]，即该行中j > k的元素）和x切片（x[k+1:]，即已计算好的后续元素），并转换为二维数组。
4. 逐行点积：用np.einsum('ij,ij->i')对每行的切片元素做逐元素相乘后求和，得到回代公式中需要减去的累加项。
5. 公式计算：代入回代公式x[i] = (c[i] - sum(R[i,j]x[j])) / R[i,i]计算并赋值。

测试验证 #

以上三角矩阵R = np.array([[2,1,3], [0,4,2], [0,0,5]])和向量c = np.array([10, 12, 15])为例，运行代码会得到正确结果x = [-0.25, 1.5, 3.0]，符合回代法的计算逻辑。

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