例如RGBA numpy 3D数组：
arr1 = numpy.random.randint(256, size = (3,3,4))
字符串[R，G，B，A]是最后一个轴中的元素。现在，当R，G和B都为0时，我将A设置为0。如何通过使用numpy where方法或任何快速方法来实现这一点？基本上，由于速度问题，我试图避免嵌套循环。
where
jckbn6z71#
在2轴上找到条件（“R、G和B均为0”）为True的行，为最后一列重新分配所需的值：
2
True
arr[(arr[:, :, :-1] == 0).all(2), -1] = 0
字符串
1条答案
按热度按时间jckbn6z71#
在
2轴上找到条件（“R、G和B均为0”）为True的行，为最后一列重新分配所需的值：字符串