代码之家 › 专栏 › 技术社区 › kmario23 Mazdak

在numpy数组中获取唯一行时保留顺序

size-reduction unique multidimensional-array numpy python

kmario23 Mazdak · 技术社区 · 7 年前

我有三个二维阵列 a1 , a2 和 a3

In [165]: a1
Out[165]: 
array([[ 0,  1,  2],
       [ 3,  4,  5],
       [ 6,  7,  8],
       [ 9, 10, 11]])

In [166]: a2
Out[166]: 
array([[ 9, 10, 11],
       [15, 16, 17],
       [18, 19, 20]])

In [167]: a3 
Out[167]: 
array([[6, 7, 8],
       [4, 5, 5]])

我把这些数组叠加成一个数组:

In [168]: stacked = np.vstack((a1, a2, a3))

In [170]: stacked 
Out[170]: 
array([[ 0,  1,  2],
       [ 3,  4,  5],
       [ 6,  7,  8],
       [ 9, 10, 11],
       [ 9, 10, 11],
       [15, 16, 17],
       [18, 19, 20],
       [ 6,  7,  8],
       [ 4,  5,  5]])

现在,我想去掉重复的行。所以, numpy.unique 做这项工作。

In [169]: np.unique(stacked, axis=0)
Out[169]: 
array([[ 0,  1,  2],
       [ 3,  4,  5],
       [ 4,  5,  5],
       [ 6,  7,  8],
       [ 9, 10, 11],
       [15, 16, 17],
       [18, 19, 20]])

然而,这里有一个问题。获取唯一行时,原始订单将丢失。如何保留原始排序并保留唯一行?

因此,预期输出应该是:

array([[ 0,  1,  2],
       [ 3,  4,  5],
       [ 6,  7,  8],
       [ 9, 10, 11],
       [15, 16, 17],
       [18, 19, 20],
       [ 4,  5,  5]])

2 回复 | 直到 7 年前

BENY 7 年前

使用 return_index

_,idx=np.unique(stacked, axis=0,return_index=True)

stacked[np.sort(idx)]
array([[ 0,  1,  2],
       [ 3,  4,  5],
       [ 6,  7,  8],
       [ 9, 10, 11],
       [15, 16, 17],
       [18, 19, 20],
       [ 4,  5,  5]])

farhad 7 年前

在得到叠加数组之后

步骤1:获取已排序的唯一数组的行索引

row_indexes = np.unique(stacked, return_index=True, axis=0)[1]

注意:行索引包含已排序数组的索引

第2步:现在用已排序的索引迭代堆栈数组

sorted_index=sorted(row_indexes)
new_arr=[]
for i in range(len(sorted_index)):
    new_arr.append(stacked[sorted_index[i]]

就是这样!!!!!!!

推荐文章

Paul Williams · 迭代数组时输出有问题

1 年前

pasty guacamole · 在结构体(C)中使用双指针访问2d数组上的信息

2 年前

Muffinman · 使用Excel O365,如何在两列上使用唯一性,对相关数据进行排序和复制?

2 年前

Suyash Nachankar · 为什么矩阵的所有行都与最后一行相同?

2 年前

Vyacheslav Verkhovin · C中二维数组的Typedef和指针上的访问

2 年前

Aditi · 我正在尝试用Java解决一个数字模式问题。我已经写了代码,如果有人能对我现有的sol进行更改,我将不胜感激

2 年前

neo · 为什么我不能访问以取消引用int指针作为索引的char指针数组中的位置

2 年前

Omprakash · 使用Google工作表中的多行和多列进行筛选

2 年前

Genki · Bash:如何从多维数组中的命名键获取值

2 年前

Insert0Name0Here · 将多维数组中的数组传递给C中的函数#

2 年前