索引NumPy数组生成含命名元组的列表时触发IndexError错误求助

错误原因 #

你的NumPy数组形状为(2143, 3)，表示它包含2143个行，每行有3个元素。你原代码中使用nz[0][i]的索引方式，是试图取第一行的第i个元素，但第一行只有3个元素，当循环变量i大于等于3时，必然触发IndexError: tuple index out of range。

正确实现方式 #

方式1：遍历数组每行，手动提取元素 #

from collections import namedtuple
import numpy as np

Voxel = namedtuple('Voxel', 'x y z c')
# nz为形状(2143, 3)的NumPy数组
voxels = [Voxel(row[0], row[1], row[2], True) for row in nz]

方式2：解包每行元素（更简洁） #

利用Python的解包语法，直接将每行的3个元素传入命名元组：

voxels = [Voxel(*row, True) for row in nz]

方式3：转为整数类型（若数组元素非整数） #

如果你的数组元素不是整数类型，需要显式转换为整数：

# 单个元素转换
voxels = [Voxel(int(row[0]), int(row[1]), int(row[2]), True) for row in nz]

# 解包时批量转换
voxels = [Voxel(*(int(x) for x in row), True) for row in nz]

验证示例 #

用小型数组测试代码逻辑：

nz = np.array([[10, 20, 30], [40, 50, 60]])
voxels = [Voxel(*row, True) for row in nz]
print(voxels[0])  # 输出：Voxel(x=10, y=20, z=30, c=True)

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