在numpy中,可以使用numpy.where()
函数来获取数组中只有一项的坐标。该函数返回一个包含满足条件的元素索引的元组。
下面是获取numpy数组中只有一项的坐标的步骤:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
numpy.where()
函数获取只有一项的坐标:index = np.where(arr == 3)
print(index)
完整的代码如下:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
index = np.where(arr == 3)
print(index)
输出结果为:
(array([2]),)
这表示在数组arr
中,只有一项的坐标是2。
对于多维数组,可以使用相同的方法来获取只有一项的坐标。下面是一个示例:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
index = np.where(arr == 5)
print(index)
输出结果为:
(array([1]), array([1]))
这表示在二维数组arr
中,只有一项的坐标是(1, 1)。
领取专属 10元无门槛券
手把手带您无忧上云