如何使用NumPy数组索引从2D数组中选择2列来从中选择唯一值?

时间:2013-04-23 20:52:14

标签: arrays numpy

这可能是一个非常愚蠢的问题,但我已经搜索过,无法在任何地方找到答案(可能是因为它太愚蠢了)。

我有一个包含多列的2D NumPy数组。我想在unique1st列中标识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]中找到所有唯一值?

1 个答案:

答案 0 :(得分:2)

在第二个维度上也使用:2切片,以包含多个列。

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