3D数组的理解列表

Question

我正在处理形状图像的 3D 数组

(225, 400, 3)

我已经完成了这个 for 循环，使图像变成黑白：

for i in imageArray[:,:,:]:
for x in i:
    if x.mean()<= 100:
        x[:] = 0
    else:
        x[:] = 255

但我无法将其列入理解列表，我尝试过：

[0 if x.mean()<=100 else 255 for x in arr_aux[:,:,0:3]]

但什么也没做

Answer 1

您将零分配给该行，而不是一行零。 正确：

[[0] * len(x) if x.mean()<=100 else [255] * len(x) for x in imageArray[:,:,0:3]]