np.nonzero()
时间: 2023-09-10 18:10:17 浏览: 70
np.nonzero()是一个NumPy函数,用于返回数组中非零元素的索引。
具体来说,它返回一个包含两个数组的元组,分别表示非零元素的行索引和列索引。这两个数组的长度等于数组中非零元素的个数。
下面是一个示例:
```python
import numpy as np
arr = np.array([[0, 1, 0],
[2, 0, 3],
[0, 4, 0]])
indices = np.nonzero(arr)
print(indices)
```
输出结果为:
```
(array([0, 1, 1, 2]), array([1, 0, 2, 1]))
```
其中,第一个数组 `[0, 1, 1, 2]` 表示非零元素的行索引,第二个数组 `[1, 0, 2, 1]` 表示非零元素的列索引。
相关问题
np.nonzero
np.nonzero函数是一个NumPy库中的方法,它用于获取数组中非零元素的索引。
这个函数的用法是:np.nonzero(array)。它的返回值是一个包含两个数组的元组,分别表示非零元素所在的行和列的索引。
举个例子,假设我们有一个数组a = np.array([0, 2, 0, 4, 0]),我们可以使用np.nonzero(a)来获取它非零元素的索引。返回的结果是一个元组(array([1, 3]),),表示非零元素在原数组中的索引位置为1和3。
另外需要注意的是,如果数组是多维的,np.nonzero()函数会返回多个数组,每个数组对应一个维度下非零元素的索引。
np.nonzero()
np.nonzero()是NumPy库中的一个函数,用于返回数组中非零元素的索引。它的作用是找到数组中非零元素的位置,并以元组的形式返回这些位置的索引。
具体来说,np.nonzero()函数接受一个数组作为参数,并返回一个包含两个数组的元组,分别表示非零元素的行索引和列索引。如果数组是多维的,则返回的行索引列索引也是多维的。
下面是一个示例:
```python
import numpy as np
arr = np.array([[0, 1, 0],
[2, 0, 3],
[0, 4, 0]])
result = np.nonzero(arr)
print(result)
```
输出结果为:
```
(array([0, 1, 1, 2]), array([1, 0, 2, 1]))
```
这表示在数组arr中,非零元素的位置分别为(0, 1)、(1, 0)、(1, 2)和(2, 1)。