我正在使用 Python3 和 numpy。我有一個像這樣的陣列：
arr = np.array([[1,2], [3,4],[5,6]])
我想找到點的位置，例如 [3,4]。我正在使用這個
idP = [id for id in range(len(arr)) if (arr[id] == [3,4]).all()]
做這個的最好方式是什么
uj5u.com熱心網友回復：
你可以這樣做：
(arr == [3, 4]).all(-1).nonzero()[0]
或通過使用np.isin：
np.isin(arr, [3, 4]).all(1).nonzero()[0]
uj5u.com熱心網友回復：
這個怎么樣？
arr.tolist().index([3,4])
uj5u.com熱心網友回復：
一種解決方案可能是將您的 numpy 陣列轉換為 python 串列
arr.tolist().index([3,4])
>>> 1
轉載請註明出處，本文鏈接：https://www.uj5u.com/qiye/464603.html