2013-04-23 63 views
2

這可能是一個非常愚蠢的問題,但我搜索並找不到任何答案(可能是因爲它太愚蠢了一個問題)。我有一個二維numpy數組與多列。我想確定第一列或第二列中的唯一元素,但不是其餘列中的元素。如何使用numpy數組索引從2D數組中選擇2列以從中選擇唯一值?

array([['A', 'B', '3', '4'], 
     ['C', 'D', '3', '5']], 
     dtype='|S1') 

使用np.unique將獲得數組中的獨特價值,我可以索引像這樣

np.unique(example_array[:,0]) 
Out[16]: 
array(['A', 'C'], 
     dtype='|S1') 

單列我怎樣才能索引,讓我可以找到[所有的獨特的價值觀;, 0]和[:,1]?

回答

2

在第二維上也使用:2切片以包含多個列。

np.unique(example_array[:,:2]) 
Out[]: 
array(['A', 'B', 'C', 'D'], 
     dtype='|S1')