如何在轴= 0周围找到最大的numpy数组，不包括0的元组？

Question

我有一个numpy数组，r，即3 x 2048 X 2048.我有以下代码来找到轴的最大值= 0。

maximum = r.max(axis=0) 

我怎样才能完成与上面相同的任务，但在找到最大值时排除0（黑色像素）的元组？

Answer 1

我理解你的问题如下：你的图像式数组可能包含一个像素值[0,0,0]，如果发生这种情况，你不希望在最大值的发现中包含它。或者换一种说法：你不希望maximum中的值为零。

如果这是你想要的，那么你可以像这样掩盖所有的零：

maximum = r.max(axis=0) # maximum has shape (2048, 2048)
maximum = maximum[maximum>0] # now maximum is 1d with length 2048^2-(number of "black pixels")