Matlab查找二维索引 - Python

在Matlab中，查找二维数组的特定值的索引是一项简单的任务，但在Python中则需要更多的代码。本文将介绍如何在Python中检索二维数组的索引，并提供代码示例和解释。

查找二维数组的索引

假设我们有以下的二维数组：

import numpy as np

arr = np.array([[0, 1, 2],
                [3, 4, 5],
                [6, 7, 8]])

在Matlab中，我们可以使用find函数来查找特定值的索引。例如，要查找数值为5的索引，可以使用以下命令：

[i,j] = find(arr == 5)

上述命令将返回一个包含2和1的i向量和一个包含3和1的j向量，表示5在数组的第2行、第3列。然而，在Python中，你需要编写更多的代码来实现相同的任务。

在Python中查找二维数组的索引

在Python中，我们可以使用numpy.where函数来查找特定值的索引。例如，要查找数值为5的索引，可以使用以下命令：

indices = np.where(arr == 5)

这将返回一个包含1和2的元组，表示5在数组的第1行、第2列。如果要将其转换为两个分开的数组，可以使用以下命令：

row_indices, col_indices = np.where(arr == 5)

上述命令将返回一个包含1的row_indices数组和一个包含2的col_indices数组。

总结

在Python中查找二维数组的索引可以使用numpy.where函数。虽然Matlab中有相应的函数可以做到同样的任务，但在Python中需要编写更多的代码。