微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

有没有办法找到一个整数偏移量,当添加到一个 numpy 数组中的每个值时,它会最大化它与另一个数组的匹配量?

如何解决有没有办法找到一个整数偏移量,当添加到一个 numpy 数组中的每个值时,它会最大化它与另一个数组的匹配量?

例如说我有 2 个列表:

a = np.array([1,3,5,6,8,9])
b = np.array([103,104,106,107,108,109])

我想要添加的值是 ~100 以添加到列表 a 作为偏移量,以匹配给定偏移量的尽可能多的列表 b 值。

我目前的解决方案基本上是

best_matches = []
for v1 in a:
    for v2 in b:
        offset = abs(v1 - v2)
        matches = np.intersect1d(a + offset,b)

        best_matches.append((offset,len(matches)))

best_offset = max(best_matches,key = lambda i : i[1]) # find offset that had the most amount of intersections when added to the first array

我需要解决方案来处理相当大的值列表,因此解决方案只需要更有效或在方法上有所不同。任何帮助表示赞赏!

解决方法

相当于没有循环的答案:

o = np.bincount(np.ravel(b[:,None] - a[None,:])).argmax()
l = len(np.intersect1d(b,(a + o)))
>>> (o,l)
(98,4)

根据评论更新 @AlexanderS.Brunmayr@Blckknght

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。