使用搜索排序的二維數組

Question

我想索引像函數：np.searchsorted([1,2,3,4,5], 3)返回2，但萬一二維：np.searchsorted([[1,2],[3,4],[5,6]], [5,6])應該返回2.我該怎麼做？

Answer 1

documentation for searchsorted表示它只適用於一維陣列。

你可以在使用內置的methds列表索引位置：

>>> a = [[1,2],[3,4],[5,6]] 
>>> a.index([5,6]) 
2 

>>> a = [[1,2],[5,6],[3,4]] 
>>> print(a.index([5,6])) 
>>> a.sort() 
>>> print(a.index([5,6])) 
1 
2 

Answer 2

a = np.array([[1,2],[3,4],[5,6]]) 
b = np.array([5,6]) 
np.where(np.all(a==b,axis=1))