我有一个值的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