这可能是一个非常愚蠢的问题,但我已经搜索过,无法在任何地方找到答案(可能是因为它太愚蠢了)。
我有一个包含多列的2D
NumPy
数组。我想在unique
或1st
列中标识2nd
个元素,但不在其余列中标识:
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]
中找到所有唯一值?
答案 0 :(得分:2)
在第二个维度上也使用:2
切片,以包含多个列。
np.unique(example_array[:,:2])
Out[]:
array(['A', 'B', 'C', 'D'],
dtype='|S1')