Python循环生成(8000,2)维度数组失败问题求助

问题根源 #

你得到的ys形状为(7921,)，主要是两个核心问题：

1. 起始索引的覆盖范围没对齐8000的总长度，且循环中未处理最后一段不足100的元素；
2. 结果追加逻辑错误，要么没按位置填充，要么把二维向量扁平化，最终导致结果变成一维列表。

修正代码方案 #

明确逻辑：每隔99个索引取起始点（0、99、198...），每个起始点对应连续100个位置（从start到start+99），执行100次矩阵乘法后填充到对应位置，最后一段不足100的也补全到8000。

完整代码示例 #

import numpy as np

# 示例输入：(8000,2)的变量x
x = np.random.rand(8000, 2)
# 替换为你实际使用的矩阵乘法权重矩阵
W = np.random.rand(2, 2)

# 初始化和x同形状的结果数组，避免列表追加的维度混乱
ys = np.zeros_like(x)

# 生成所有起始索引：0,99,198,...直到不超过8000
start_indices = np.arange(0, 8000, 99)

for start in start_indices:
    # 取出当前起始点的x值
    current_obs = x[start].copy()
    # 存储当前段的100次乘法结果
    segment_results = []
    
    # 执行100次矩阵乘法
    for _ in range(100):
        current_obs = W @ current_obs
        segment_results.append(current_obs)
    
    # 转换为numpy数组，形状为(100,2)
    segment_results = np.array(segment_results)
    # 计算当前段的结束索引，不超过总长度8000
    end_idx = min(start + 100, 8000)
    # 填充到ys的对应位置，最后一段不足100则取对应长度的结果
    ys[start:end_idx] = segment_results[:end_idx - start]

# 验证结果形状，输出应为(8000,2)
print(ys.shape)

关键修正点说明 #

• 用数组初始化替代列表追加：直接创建和x同形状的零数组，按位置填充，彻底避免列表追加导致的维度混乱；
• 处理边界情况：最后一个起始点7920到8000只有80个元素，通过min(start+100,8000)和切片segment_results[:end_idx-start]确保不会越界；
• 避免扁平化：每次矩阵乘法的结果是(2,)的向量，存储到列表后转成(100,2)的数组，再填充到ys的对应切片，保证维度始终正确。

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