在 numpy 数组中查找具有指定 x 值的点的 y 坐标的最佳方法是什么
What is the best way to find y coordinates of point with specified x value in numpy array
我正在使用 python 一个 numpy 数组。我想找到具有特定 x 坐标的点的所有 y 坐标。我用这个:
[it[1] for it in arrP if it[0] == specX]
有没有更好的方法?
arr = np.array([[1,2],[1,3],[2,3]])
# select x in arr with x[0] == 1, and slice out x[1]
arr[arr[:, 0] == 1][:, 1]
产出
array([2, 3])
我正在使用 python 一个 numpy 数组。我想找到具有特定 x 坐标的点的所有 y 坐标。我用这个:
[it[1] for it in arrP if it[0] == specX]
有没有更好的方法?
arr = np.array([[1,2],[1,3],[2,3]])
# select x in arr with x[0] == 1, and slice out x[1]
arr[arr[:, 0] == 1][:, 1]
产出
array([2, 3])