本文介绍如何在 Python 中对包含不同维度 NumPy 数组的列表进行动态索引与赋值，通过构建可变长度的索引元组，实现对任意组合维度（如 n_m1 个 m1 和 n_m2 个 m2）的精准写入。
本文介绍如何在 python 中对包含不同维度 numpy 数组的列表进行动态索引与赋值，通过构建可变长度的索引元组，实现对任意组合维度（如 `n_m1` 个 `m1` 和 `n_m2` 个 `m2`）的精准写入。
在科学计算和张量操作中，我们常需处理一组维度各异的数组（例如不同阶的张量），并根据运行时参数动态定位并修改其元素。直接硬编码索引（如 arr[i, j, k]）无法满足 n_m1 和 n_m2 作为变量的场景。关键在于：*NumPy 数组支持元组索引（tuple indexing），而 Python 元组可通过 `重复操作和+` 拼接灵活构造**。
以下是一个完整、健壮的实现方案：
import numpy as np
# 初始化示例：list[i] 是 i 维的 (m, m, ..., m) 数组
m = 10
i_range = 6
arrays = [np.zeros((m,) * i) for i in range(i_range)] # 注意：避免用 list 作变量名（覆盖内置类型）
# 核心写入函数
def write_to_jagged(arrays, *, i, m1, m2, n_m1, n_m2, val):
"""
向 arrays[i] 的指定位置写入值 val。
参数：
- arrays: 包含不同维度 NumPy 数组的列表
- i: 数组索引（即选择第 i 个数组）
- m1, m2: 索引值
- n_m1, n_m2: 对应索引值的重复次数（要求 n_m1 + n_m2 == arrays[i].ndim）
- val: 待写入的标量或兼容形状的值
"""
if not (0 <p>✅ <strong>关键要点说明：</strong> </p>
-
元组索引是核心：
arr[(a,b,c)]等价于arr[a,b,c]，因此(m1,)*n_m1 + (m2,)*n_m2可生成任意长度的合法索引元组； -
维度校验很重要：务必确保
n_m1 + n_m2 == arrays[i].ndim，否则将触发IndexError或静默越界（取决于数组形状）； -
命名规范提醒：示例中将原
list改为arrays，避免覆盖 Python 内置list类型，提升代码健壮性与可读性； -
扩展性友好：该模式可轻松推广至
k种索引值（如(m1,)*n1 + (m2,)*n2 + (m3,)*n3），无需修改底层逻辑。
掌握这一技巧后，你即可在循环、递归或配置驱动的张量操作中，安全、高效地完成动态维度索引，大幅提升数值代码的灵活性与复用性。
