2019-04-16
阅读量:
781
从[1,2,0,0,4,0]查找非零元素的索引
arr = np.array([1,2,0,0,4,0])
# list comprehension is not consise VS nonzero
nz1 = [i for i in range(len(arr)) if arr[i]==0] # a list
nz = np.nonzero(arr) # return a tuple
print nz
print nz1
(array([0, 1, 4]),)
[2, 3, 5]
0.0000
0
1
关注作者
收藏
评论(0)
发表评论
暂无数据
