NUMPY010 题解：找出非零元素索引

题目

找出数组 [1,2,0,0,4,0] 中所有非零元素的索引。

解题思路

使用 np.nonzero() 函数返回非零元素的索引。

代码

import numpy as np
nz = np.nonzero([1, 2, 0, 0, 4, 0])
print(nz[0])

代码详解

1. np.nonzero([1, 2, 0, 0, 4, 0]) —— 返回非零元素的索引
2. 返回的是一个元组，元组第一个元素是索引数组
3. 输出：[0 1 4]

索引含义

索引:  0   1   2   3   4   5
值:   [1   2   0   0   4   0]
       ↑   ↑           ↑
      非零            非零

• 索引 0 的值是 1（非零）
• 索引 1 的值是 2（非零）
• 索引 4 的值是 4（非零）

其他方法

# 使用 np.where
np.where(arr != 0)[0]

核心知识点

1. np.nonzero() —— 返回非零元素索引
2. 返回结果是元组
3. [0] 取第一维的索引