从 numpy 数组中连接并删除重复元素

Question

我在循环的每次迭代中计算一个索引数组，然后我想删除重复的元素并将计算的数组连接到前一个数组。 例如，第一次迭代给了我这个数组：

array([  1,   6,  56, 120, 162, 170, 176, 179, 197, 204])

第二个：

array([ 29,  31,  56, 104, 162, 170, 176, 179, 197, 204]) 

等等。我该怎么办？

Answer 1

您可以先使用numpy.concatenate 连接数组，然后使用np.unique

import numpy as np
a=np.array([1,6,56,120,162,170,176,179,197,204])
b=np.array([29,31,56,104,162,170,176,179,197,204])
new_array = np.unique(np.concatenate((a,b),0))

print new_array

结果：

[  1   6  29  31  56 104 120 162 170 176 179 197 204]

Answer 2

您可以使用numpy.concatenate 和numpy.unique：

In [81]: arr = np.array([  1,   6,  56, 120, 162, 170, 176, 179, 197, 204])

In [82]: arr = np.unique(np.concatenate((arr, np.array([ 29,  31,  56, 104, 162, 170, 176, 179, 197, 204]))))

In [83]: arr
Out[83]: array([  1,   6,  29,  31,  56, 104, 120, 162, 170, 176, 179, 197, 204])