numpy使用技巧记录

1.查找数组中不重复的部分

>>a = np.array([[1,2,3,8,7,5,6],[10,2,38,81,0,3,12],[-10,5,39,1,19,25,3],[10,-2,15,20,12,36,10],[0,2,8,-5,0,20,86]])
>>np.unique(a[:,0])
>>[-10   0   1  10]

2.按照列查找多维数据

>>a = np.array([[1,2,3,8,7,5,6],[10,2,38,81,0,3,12],[-10,5,39,1,19,25,3],[10,-2,15,20,12,36,10],[0,2,8,-5,0,20,86]])
>>print(np.lexsort(a.T[0,None]))
>>[2 4 0 1 3]