本文介绍如何在 Python 中对包含不同维度 NumPy 数组的列表进行动态索引与赋值，尤其适用于维度数由变量（如 n_m1 和 n_m2）决定的场景，通过构造元组索引实现灵活、安全、高效的元素写入。
本文介绍如何在 python 中对包含不同维度 numpy 数组的列表进行动态索引与赋值，尤其适用于维度数由变量（如 `n_m1` 和 `n_m2`）决定的场景，通过构造元组索引实现灵活、安全、高效的元素写入。
在科学计算和数值模拟中，我们常需处理一组结构不统一的多维数组——例如，第 i 个数组是 i 维的超立方体（如 shape = (m,) * i）。此时，若需向特定位置（如前 n_m1 个轴取索引 m1，后 n_m2 个轴取索引 m2，且满足 n_m1 + n_m2 == i）写入值，硬编码索引（如 [m1, m1, m2]）不再可行，必须借助程序化方式构造索引。
核心思路非常简洁：NumPy 数组支持元组索引（tuple indexing），而 Python 元组可通过重复操作符 * 和拼接操作符 + 动态生成。例如 (m1,) * 3 得到 (m1, m1, m1)，(m1,) * 2 + (m2,) * 3 得到 (m1, m1, m2, m2, m2)。结合列表索引获取目标数组，即可完成一次完整赋值。
以下是一个健壮、可复用的封装函数：
import numpy as np
def write_to_jagged_array(jagged_list, *, i, m1, m2, n_m1, n_m2, value):
"""
向 jagged_list[i] 的指定位置写入值。
参数:
jagged_list: list of np.ndarray，各元素维度依次递增
i: 列表索引，对应目标数组（其维度为 i）
m1, m2: 用于索引的整数值（需在各自轴的有效范围内）
n_m1, n_m2: 分别表示 m1 和 m2 在索引元组中出现的次数，要求 n_m1 + n_m2 == i
value: 待写入的标量或兼容形状的值
"""
if not (0 <p>使用示例（基于问题中的初始化逻辑）：</p><pre class="brush:php;toolbar:false;">m = 10
i_range = 6
jagged = [np.zeros((m,) * i) for i in range(i_range)]
# 向第 4 个数组（即 4 维数组，shape=(10,10,10,10)）写入：
# 前 2 个轴取索引 3，后 2 个轴取索引 7 → 即位置 [3,3,7,7]
write_to_jagged_array(jagged, i=4, m1=3, m2=7, n_m1=2, n_m2=2, value=99.0)
print(jagged[4][3, 3, 7, 7]) # 输出: 99.0⚠️ 注意事项：
-
jagged[i]必须是 NumPy 数组；若为标量（如i=0时的 0 维数组），index_tuple应为空元组()，此时(m1,)*0 + (m2,)*0 == ()，仍可正确赋值（等价于arr[()] = value）。 - 确保
m1和m2在各自轴的合法范围内（0 ），否则将触发 <code>IndexError。 - 避免使用
list作为变量名（如原示例），它会覆盖内置类型list；推荐改用jagged、arrays或ndarray_list等名称。 - 该方法完全基于 NumPy 原生索引机制，无额外开销，性能优异。
总结而言，利用 Python 元组的动态构造能力 + NumPy 的元组索引语法，即可优雅解决“变量维度数组的通用写入”这一常见需求，无需反射、eval 或复杂递归，代码清晰、安全且高效。
