如何获取numpy数组中重复元素的所有索引的列表

问题描述：

我正在尝试获取numpy数组中所有重复元素的索引，但是我目前发现的解决方案对于大型(> 20000个元素)输入数组而言效率非常低(大约需要9秒钟的时间)).这个想法很简单:

I'm trying to get the index of all repeated elements in a numpy array, but the solution I found for the moment is REALLY inefficient for a large (>20000 elements) input array (it takes more or less 9 seconds). The idea is simple:

records_array 是一个时间戳的numpy数组( datetime )，我们要从中提取重复时间戳的索引

records_arrayis a numpy array of timestamps (datetime) from which we want to extract the indexes of repeated timestamps

time_array 是一个numpy数组，包含在 records_array

time_array is a numpy array containing all the timestamps that are repeated in records_array

records 是一个django QuerySet(可以轻松转换为列表)，其中包含一些Record对象.我们要创建一个由Record的tagId属性的所有可能组合形成的对的列表，这些对应该与从 records_array 找到的重复时间戳相对应.

records is a django QuerySet (which can easily converted to a list) containing some Record objects. We want to create a list of couples formed by all possible combinations of tagId attributes of Record corresponding to the repeated timestamps found from records_array.

这是我目前拥有的有效(但效率低下)的代码:

Here is the working (but inefficient) code I have for the moment:

tag_couples = [];
for t in time_array:
    users_inter = np.nonzero(records_array == t)[0] # Get all repeated timestamps in records_array for time t
    l = [str(records[i].tagId) for i in users_inter] # Create a temporary list containing all tagIds recorded at time t
    if l.count(l[0]) != len(l): #remove tuples formed by the first tag repeated
        tag_couples +=[x for x in itertools.combinations(list(set(l)),2)] # Remove duplicates with list(set(l)) and append all possible couple combinations to tag_couples

我非常确定可以使用Numpy对其进行优化，但是我找不到一种方法，不用使用for就能将 records_array 与 time_array 的每个元素进行比较循环(不能同时使用 == 进行比较，因为它们都是数组).

I'm quite sure this can be optimized by using Numpy, but I can't find a way to compare records_array with each element of time_array without using a for loop (this can't be compared by just using ==, since they are both arrays).

答

使用numpy的矢量化解决方案，具有

A vectorized solution with numpy, on the magic of unique().

import numpy as np

# create a test array
records_array = np.array([1, 2, 3, 1, 1, 3, 4, 3, 2])

# creates an array of indices, sorted by unique element
idx_sort = np.argsort(records_array)

# sorts records array so all unique elements are together 
sorted_records_array = records_array[idx_sort]

# returns the unique values, the index of the first occurrence of a value, and the count for each element
vals, idx_start, count = np.unique(sorted_records_array, return_counts=True, return_index=True)

# splits the indices into separate arrays
res = np.split(idx_sort, idx_start[1:])

#filter them with respect to their size, keeping only items occurring more than once
vals = vals[count > 1]
res = filter(lambda x: x.size > 1, res)

以下代码是原始答案，需要使用 numpy 广播并两次调用 unique 两次，这需要更多的内存:

The following code was the original answer, which required a bit more memory, using numpy broadcasting and calling unique twice:

records_array = array([1, 2, 3, 1, 1, 3, 4, 3, 2])
vals, inverse, count = unique(records_array, return_inverse=True,
                              return_counts=True)

idx_vals_repeated = where(count > 1)[0]
vals_repeated = vals[idx_vals_repeated]

rows, cols = where(inverse == idx_vals_repeated[:, newaxis])
_, inverse_rows = unique(rows, return_index=True)
res = split(cols, inverse_rows[1:])

具有预期的 res = [array([0，3，4])，array([1，8])，array([2，5，7])]

如何获取numpy数组中重复元素的所有索引的列表

相关推荐