Python中基于嵌套循环堆叠矩阵结果的实现问题求助

首先看你代码里的几个核心问题：

• 缩进错误：最后一行z_weights_time[s,:]=np.vstack(z_weights)没缩进，不在外层for s循环里，循环结束后s变成6，而z_weights_time只有6行（索引0-5），直接赋值会触发索引越界。
• 内层循环逻辑错误：你把z=y*c和z_weights=z.sum(axis=0)放在了内层for i循环里，这会导致每次i迭代都覆盖z和z_weights，最后只保留了i=3时的结果，正确的做法应该是先完成整个y矩阵的构建，再计算z和z_weights。

修正后的循环版本代码 #

import numpy as np

x = np.array([[1,2,3,4],[2,2,4,4],[0,5,3,4],[1,2,1,4],[1,0,3,4],[4,2,3,3]])
c = np.random.rand(4,4)
z_weights_time = np.zeros((6,4))

for s in range(6):
    d = x[s, :]  # 直接取第s行，不用x[[s],:]，得到一维数组更方便
    y = np.zeros((4,4))
    for i in range(4):
        y[i, :] = d[i] - d  # 一维数组直接运算，无需vstack
    z = y * c  # 元素级相乘，对应你的步骤3
    z_weights = z.sum(axis=0)
    z_weights_time[s, :] = z_weights  # 缩进后放入外层循环内

print("z_weights_time:\n", z_weights_time)

更高效的广播版本（避免循环） #

numpy的广播特性可以帮你彻底去掉嵌套循环，代码更简洁且运行速度更快：

import numpy as np

x = np.array([[1,2,3,4],[2,2,4,4],[0,5,3,4],[1,2,1,4],[1,0,3,4],[4,2,3,3]])
c = np.random.rand(4,4)

# 利用广播生成所有行对应的y矩阵，最终维度为(6,4,4)
y = x[:, np.newaxis, :] - x[:, :, np.newaxis]
# 元素级相乘后按列求和：先得到(6,4,4)的结果，再沿axis=1求和得到(6,4)的最终矩阵
z_weights_time = (y * c).sum(axis=1)

print("z_weights_time:\n", z_weights_time)

广播逻辑解释：

• x[:, np.newaxis, :]将原x的维度转换为(6,1,4)
• x[:, :, np.newaxis]将原x的维度转换为(6,4,1)
• 两者相减时自动广播为(6,4,4)，对应每行x生成一个4x4的y矩阵
• 和维度为(4,4)的c相乘后，沿axis=1求和，直接得到每行对应的1x4结果，自动堆叠为6x4矩阵

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