【Python】获取数组中⾮零元素
第⼀种⽅法flatnonzero
import numpy as np
a = np.array([[1,2,3],[4,5,6]])
b = np.array([[True,True,False],[False,True,False]])
c = a*b
print(c.ravel()[np.flatnonzero(c)])
out:
[1 2 5]
第⼆种⽅法nonzero
nonzero(a)
返回数组a中⾮零元素的索引值数组。
（1）只有a中⾮零元素才会有索引值，那些零值元素没有索引值；
（2）返回的索引值数组是⼀个2维tuple数组，该tuple数组中包含⼀维的array数组。其中，⼀维array向量的个数与a的维数是⼀致的。
（3）索引值数组的每⼀个array均是从⼀个维度上来描述其索引值。⽐如，如果a是⼀个⼆维数组，则索引值数组有两个array，第⼀个array从⾏维度来描述索引值；第⼆个array从列维度来描述索引值。
（4） 该np.o(x))
函数能够描述出每⼀个⾮零元素在不同维度的索引值。
（5）通过a[nonzero(a)]得到所有a中的⾮零值python获取数组长度
#a是1维数组
a = [0,2,3]
b = np.nonzero(a)
print(np.array(b).ndim)
print(b)
结果：
2
(array([1, 2], dtype=int64),)
说明：索引1和索引2的位置上元素的值⾮零。
#a是2维数组
a = np.array([[0,0,3],[0,0,0],[0,0,9]])
b = np.nonzero(a)
print(np.array(b).ndim)
print(b)
o(a)))
结果：
2
(array([0, 2], dtype=int64), array([2, 2], dtype=int64))
[[0 2]
[2 2]]
说明：
（1）a中有2个⾮零元素，因此，索引值tuple中array的长度为2。因为，只有⾮零元素才有索引值。
（2）索引值数组是2 维的。实际上，⽆论a的维度是多少，索引值数组⼀定是2维的tuple，但是tuple中的⼀维array个数和a的维数⼀致。
（3）第1个array([0, 2])是从row值上对3和9进⾏的描述。第2个array([2, 2])是从col值上对3和9的描述。这样，从⾏和列上两个维度上各⽤⼀个数组来描述⾮零索引值。
（4）通过调⽤np.transpose()函数，得出3的索引值是[0 2]，即第0⾏，第2列。
#a是3维数组
a = np.array([[[0,0],[1,0]],[[0,0],[1,0]],[[0,0],[1,0]]])
b = np.nonzero(a)
print(np.array(b).ndim)
print(b)
结果：
2
(array([0, 1, 2], dtype=int64), array([1, 1, 1], dtype=int64), array([0, 0, 0], dtype=int64))
说明：由于a是3维数组，因此，索引值数组有3个⼀维数组。
注：数组索引值从0开始。

版权声明：本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198，我们将在24小时内删除。