np.unravel_index
时间: 2023-12-03 14:43:48 浏览: 236
matlab.rar_unravel matlab_图像处理_图形增强_图形处理
`np.unravel_index`是NumPy中的一个函数,用于将一个一维数组中的索引转换为多维数组中的索引。例如,如果我们有一个形状为 `(3, 4)` 的数组,并且想要将索引 `6` 转换为二维索引 `(1, 2)`,我们可以使用 `np.unravel_index`:
```python
import numpy as np
arr = np.arange(12).reshape((3, 4))
print(arr)
# Output:
# [[ 0 1 2 3]
# [ 4 5 6 7]
# [ 8 9 10 11]]
index = 6
index_2d = np.unravel_index(index, arr.shape)
print(index_2d)
# Output: (1, 2)
```
在上面的例子中,`np.unravel_index` 将索引 `6` 转换为二维索引 `(1, 2)`,即第二行第三列。`np.unravel_index` 还可以接受一个数组作为参数,将数组中的每个元素的索引转换为多维数组中的索引。
阅读全文