本文详细介绍了 NumPy 中的 nonzero 函数，该函数用于返回数组中非零元素的索引。通过一维、二维及三维数组的例子，展示了如何使用 nonzero 函数获取不同维度下非零元素的位置，为数据处理和分析提供了实用的技巧。
nonzero：返回非零元素的索引。 返回数组的元组，每个数组对应一个维度，其中包含该维度中非零元素的索引。
# 一维
a = [1, 0, 5]
b = nonzero(a)
print(b)
# 二维
a = array([[0, 0, 1], [2, 0, 0], [0, 6, 0]])
b = nonzero(a)
print(b)
# 三维
a = array([[[2, 0], [1, 0]], [[0, 1], [0, 0]], [[0, 0], [0, 5]]])
b = nonzero(a)
print(b)
扫一扫
3769
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
