gpt4 book ai didi

python - 在python中查找两个数组之间互斥元素的索引

转载 作者:太空宇宙 更新时间:2023-11-03 17:12:21 25 4
gpt4 key购买 nike

我有两个数组,并且我已经找到了如何使用 np.setxor1d(a,b) 识别互斥元素。例如:

a = np.random.randint(11, size=10) #first array
b = np.random.randint(11, size=10) #second array
ex = np.setxor1d(a,b) #mutually exclusive array

a
Out[1]: [1, 5, 3, 7, 6, 0, 10, 10, 0, 9]
b
Out[2]: [1, 9, 8, 6, 3, 5, 8, 0, 3, 10]
ex
Out[3]: [7, 8]

现在,我试图弄清楚如何获取 ab 的独占数组 ex 的元素索引>。以诸如a_mutex_indb_mutex_ind之类的方式。有谁知道不使用 for 循环的聪明方法吗?谢谢!

最佳答案

>>> x = np.setxor1d(a, b)
>>> i, = np.nonzero(np.in1d(a, x))
>>> i
array([3])
>>> a[i]
array([7])

对于b也类似:

>>> j, = np.nonzero(np.in1d(b, x))
>>> j
array([2, 6])
>>> b[j]
array([8, 8])

关于python - 在python中查找两个数组之间互斥元素的索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/33975665/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com