如何提取3D NumPy数组中的每一列数据

你当前使用的3D数组维度结构为(n_slice, n_row, 3)，其中最后一维对应CIELab的3个通道值，需要提取的「列」就是最后一维指定索引对应的全量数值，以下是可直接运行的实现方案：

核心实现逻辑 #

你给出的示例数组形状为(3,5,3)，对应3个二维切片、每个切片5行、每行3个通道值。示例中展示的第一列提取结果，本质是取所有位置上最后一维索引为0的数值，按切片顺序逐行排列即可。

方案1：通道优先转换（最适配后续Lab公式计算） #

针对CIELab三通道计算的场景，直接把通道维度移动到数组最前，后续可以直接按索引取L、a、b三个通道的全量值，不需要重复写索引逻辑：

import numpy as np

# 输入的示例3D数组
lab_arr = np.array([
    [[ 65,234,169],[203,191,245],[ 36, 58,196],[207,208,143],[251,208,187]],
    [[ 79, 69,237],[ 13,124, 42],[104,165, 82],[170,178,178],[ 66, 42,210]],
    [[ 40,163,219],[142, 37,140],[ 75,205,143],[246, 30,221],[ 16, 98,102]]
])

# 将通道维度移至最前，输出形状为(3, 3, 5)，对应[通道数, 切片数, 单切片行数]
lab_channels = np.moveaxis(lab_arr, source=-1, destination=0)

# 提取第一列（L通道，索引为0）
L_channel = lab_channels[0]

执行后L_channel的存储结果为：

array([[ 65, 203,  36, 207, 251],
       [ 79,  13, 104, 170,  66],
       [ 40, 142,  75, 246,  16]])

如果需要输出成示例中带空行分隔的逐行打印格式，加一段遍历逻辑即可：

for slice_block in L_channel:
    for value in slice_block:
        print(f"{value:>10}")
    print()

输出效果和期望格式完全一致。
如果需要提取a、b通道，直接取lab_channels[1]、lab_channels[2]即可，数值顺序和原像素位置完全对应，可直接代入后续公式计算。

方案2：单列表直接索引（轻量场景用） #

如果只需要临时提取某一列，不需要拆分全部通道，直接用多维索引即可，不需要做轴变换：

# 提取第1列（索引0）
first_col = lab_arr[:, :, 0]
# 提取第2列（索引1）
second_col = lab_arr[:, :, 1]
# 提取第3列（索引2）
third_col = lab_arr[:, :, 2]

返回结果和方案1完全一致。

注意事项 #

• 用skimage、OpenCV等库转换得到的CIELab数组默认都是通道在最后一维的结构，上述方法提取的通道值不会打乱像素映射关系，可直接使用。
• 如果你的数组维度顺序有调整，只需要修改moveaxis里的源轴参数，匹配自己的通道所在维度即可。

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