import numpy as np ''' nonzero() 获取元素的行下标与列下标 ''' a = np.array([[1, 0, 3], [4, 5, 0]]) print(np.nonzero(a)) ''' 输出结果: (array([0, 0, 1, 1]), array([0, 2, 0, 1])) array([0, 0, 1, 1]) 代表a所在的行标号 ,从第一行,第二行,依次进行非零的元素下标 array([0, 2, 0, 1] 代表a所在的列标号 ,从第一列,第二列,依次进行非零的元素下标 ''' # 应用 输出a中所有元素等于5所在的行与列 print(np.nonzero(a == 5)) # (array([1]), array([1]))
numpy中nonzero函数详解
猜你喜欢
转载自blog.csdn.net/qq_30638831/article/details/80481709
今日推荐
周排行