如何修复循环中出现的IndexError: invalid index to scalar variable错误？

错误根源 #

你遇到的IndexError: invalid index to scalar variable.是因为：

• 当前写法中angle1 = np.deg2rad(np.random.uniform(0, 10))返回的是单个标量值（不是数组），而interchange_indices是长度为i的布尔数组，用数组去索引标量必然报错——标量没有索引维度。
• 你最初的写法angle1 = np.deg2rad(np.random.uniform(0, 10, i))是正确的逻辑，它会生成长度为i的数组，能和interchange_indices匹配。你提到的“i的随机采样存在问题”，大概率是指i=0时生成了空数组，引发后续逻辑报错。

修正方案 #

方案1：处理i=0的边界情况（保留数组生成逻辑） #

在循环中加入i>0的判断，仅当数组长度有效时执行交换操作：

for i in range(N):
    # 生成长度为i的数组，i=0时为空数组
    angle1 = np.deg2rad(np.random.uniform(0, 10, i))
    angle2 = np.deg2rad(np.random.uniform(0, 180, i))
    
    # 仅当i>0时执行交换，避免空数组/标量索引问题
    if i > 0:
        interchange_indices = np.random.rand(i) < 0.5
        angle1[interchange_indices], angle2[interchange_indices] = angle2[interchange_indices], angle1[interchange_indices]
    
    kappa = evolve(angle1, angle2, L)

方案2：直接跳过i=0的循环（如果evolve无法处理空数组） #

如果evolve函数不能处理空数组，可直接跳过i=0的迭代：

for i in range(N):
    if i == 0:
        # 处理i=0的场景，比如赋值默认值或跳过
        kappa = 0  # 替换为你的默认值逻辑
        continue
    
    angle1 = np.deg2rad(np.random.uniform(0, 10, i))
    angle2 = np.deg2rad(np.random.uniform(0, 180, i))
    
    interchange_indices = np.random.rand(i) < 0.5
    angle1[interchange_indices], angle2[interchange_indices] = angle2[interchange_indices], angle1[interchange_indices]
    
    kappa = evolve(angle1, angle2, L)

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