在numpy 3-D数组中找到零值

时间:2019-06-06 02:45:20

标签: python numpy

我有一个值的np 3D数组:

a = np.array([[[ 1,  2,  3],
         [ 0,  5,  6]],

        [[ 7,  0,  9],
        [10, 0, 12]],

        [[ 0, 13, 0],
        [-1, 15, 0]]])

您可以说这个数组由3帧组成,每帧是(2 * 3)数组

我想确定此数组中每行零值的位置:

我的期望是像这样的2D数组:

OUT=[[2, 1, 2]
    ,[0, 1, 2]]

此数组确定沿帧号轴的每一行中值为零的帧。

您能帮我做到这一点吗? np.where效果很好,但仅适用于2D数组,但是这里我的数组是3D

0 个答案:

没有答案