Python中如何用for循环替换3D数组指定列的元素？

你遇到的核心问题是循环中没有定位到单个像素位置，导致每次迭代都把整个通道的所有元素替换成当前循环变量，最终所有位置都被最后一次循环的值覆盖。结合你的需求（U尺寸为3×154×184），下面给出两种解决方案：

一、修正循环写法（针对习惯用循环的场景） #

如果必须用for循环，要确保每次循环只给U的单个像素位置赋值，而不是整列/整通道。示例代码如下：

import numpy as np

# 初始化3×154×184的全0数组
U = np.zeros((3, 154, 184), dtype=np.float64)
# 假设A是154×184的输入像素数组
A = np.random.rand(154, 184)

# 遍历每个像素的行(i)和列(j)
for i in range(154):
    for j in range(184):
        # 将A的(i,j)像素值赋值给U的第二通道（索引1）对应位置
        U[1, i, j] = A[i, j]
        # 第三通道（索引2）可按需赋值，比如当前列索引j
        U[2, i, j] = j

注意：NumPy是0索引，所以"第二列"对应第一个维度的索引1，"第三列"对应索引2。

二、矢量化操作（推荐，效率远高于循环） #

NumPy的优势在于矢量化运算，完全不需要循环就能实现批量赋值，代码更简洁且速度更快：

import numpy as np

U = np.zeros((3, 154, 184), dtype=np.float64)
A = np.random.rand(154, 184)

# 直接给整个第二通道赋值A
U[1, :, :] = A
# 生成第三通道的列索引值（矢量化方式）
U[2, :, :] = np.tile(np.arange(184), (154, 1)).T

适配Matlab索引习惯的小技巧 #

Matlab的数组索引是(行, 列, 通道)，而你当前的U是(通道, 行, 列)。如果习惯Matlab的顺序，可以把U初始化为(154, 184, 3)，这样索引逻辑和Matlab完全一致：

# Matlab风格的数组形状：154行×184列×3通道
U = np.zeros((154, 184, 3))
# 第二通道赋值
U[:, :, 1] = A
# 第三通道赋值
U[:, :, 2] = np.arange(184)

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