2

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

我有一个2D NumPy包含多列的数组。我想识别or列unique中的元素,而不是其余列中的元素:1st2nd

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')

我怎样才能索引它,以便我可以找到 in[;,0]和 in 中的所有唯一值[:,1]

4

1 回答 1

2

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

np.unique(example_array[:,:2])
Out[]: 
array(['A', 'B', 'C', 'D'], 
      dtype='|S1')
于 2013-04-23T20:55:00.273 回答