Question

我有三个排序的数组：

>>> a = arange(10)
>>> b = arange(3,12)
>>> c = arange(-2,8)
>>> print(a)
[0 1 2 3 4 5 6 7 8 9]
>>> print(b)
[ 3  4  5  6  7  8  9 10 11]
>>> print(c)
[-2 -1  0  1  2  3  4  5  6  7]

我想获得所有其他数组中包含的每个数组元素索引的列表。

在这个例子中，每个数组中的索引对应于数字3 - 7

如下所示：

a_inds, b_inds, c_inds = get_unq_inds(a,b,c)

a_inds = [3,4,5,6,7] (or [False, False, False,  True,  True,  True,  True,  True,  False,  False])

b_inds = [0,1,2,3,4] (or [True, True, True, True, True, False, False, False, False])

等

基本上，我想扩展此处提供的解决方案：（Find indices of common values in two arrays）到3个数组。（或者，如果你有野心勃勃，'n'阵列）

Answer 1

你可以这样做：

add_executable(myapp main.cpp)
target_link_libraries(myapp test)

<强>输出

def get_unq_inds(a, b, c):
    uniq_vals = list(set(a).intersection(b).intersection(c))
    return [a.index(x) for x in uniq_vals], [b.index(x) for x in uniq_vals], [c.index(x) for x in uniq_vals]
    # you can use this for boolean values
    #return [x in uniq_vals for x in a], [x in uniq_vals for x in b], [x in uniq_vals for x in c]

布尔值的

OUTPUT ：

a_inds, b_inds, c_inds = get_unq_inds(range(9), range(3,12), range(-2,8))

>>> a_inds, b_inds, c_inds
([3, 4, 5, 6, 7], [0, 1, 2, 3, 4], [5, 6, 7, 8, 9])

现场演示here

Answer 2

对于布尔值列表，您可以使用列表推导。

a_inds = [x in b and x in c for x in a]

Answer 3

您可以组合所有范围以获取set中的公共元素，然后对此进行测试：

>>> ranges = [range(9), range(3,12), range(-2,8)]
>>> s = set.intersection(*map(set,ranges))
>>> [[i for i,x in enumerate(sublist) if x in s] for sublist in ranges]
[[3, 4, 5, 6, 7], [0, 1, 2, 3, 4], [5, 6, 7, 8, 9]]

这适用于任意数量的输入列表。

或类似地使用与@Ashish Ranjan的想法相同的s（请注意，索引可能不会被排序，因为我们正在迭代无序set()，尽管在实践中它们很可能由于Python哈希整数的方式维护顺序：

[[sublist.index(x) for x in s] for sublist in ranges]

Python：获取3个排序数组中常见值的索引

3 个答案: