使用np.nonzero返回类型为uint16的索引值
时间: 2024-02-05 14:12:47 浏览: 106
字符型接口特性及时序-python求numpy中array按列非零元素的平均值案例
可以使用astype()方法将返回的索引值类型转换为uint16。示例如下:
```python
import numpy as np
arr = np.array([[1, 0, 3], [4, 5, 0], [0, 7, 8]])
# 返回类型为uint16的索引值
indices = np.nonzero(arr)
# 将索引值类型转换为uint16
indices = np.array(indices).astype(np.uint16)
print(indices)
```
输出结果为:
```
[[0 0 0 1 1 2 2]
[0 2 3 0 1 1 2]], dtype=uint16)
```
阅读全文