python - 从另一个数组中匹配位置的数组和元素中删除重复项

Question

我有两个 numpy 数组，我想从第一个数组中删除重复值(包括原始值)并删除第二个数组中匹配位置的项目。

例如:

a = [1, 2, 2, 3]
b = ['a', 'd', 'f', 'c']

变成:

a = [1, 3]
b = ['a', 'c']

我需要有效地做到这一点，而不是使用耗时的天真解决方案

Answer 1

这是一个 np.unique -

unq,idx,c = np.unique(a, return_index=True, return_counts=True)
unq_idx = np.sort(idx[c==1])
a_out = a[unq_idx]
b_out = b[unq_idx]

sample 运行-

In [34]: a
Out[34]: array([1, 2, 2, 3])

In [35]: b
Out[35]: array(['a', 'd', 'f', 'c'], dtype='|S1')

In [36]: unq,idx,c = np.unique(a, return_index=1, return_counts=1)
    ...: unq_idx = idx[c==1]
    ...: a_out = a[unq_idx]
    ...: b_out = b[unq_idx]

In [37]: a_out
Out[37]: array([1, 3])

In [38]: b_out
Out[38]: array(['a', 'c'], dtype='|S1')