np.flatnonzero(a) 1 参数: a:ndarray,输入数组。 返回: res:ndarray,输出数组,包含a.ravel()中非零元素的索引。 例子: >>> x = np.arange(-2, 3) >>> x array([-2, -1, 0, 1, 2]) >>> np.flatnonzero(x) array([0, 1, 3, 4]) # 使用非零元素的索引作为索引数组来提取这些元素 >>> x.ravel()[np.flatnonzero(x)] array([-2, -1, 1, 2])