gpt4 book ai didi

python - 比较两个不同大小的 numpy 数组的最佳方法

转载 作者:太空宇宙 更新时间:2023-11-04 09:23:46 28 4
gpt4 key购买 nike

<分区>

我有 2 个不同大小的 numpy 数组。从理论上讲,一个将是另一个的子集。我想在较大的 numpy 数组 中找到索引,其中它的值与较小的子集匹配。

例如

A = [ 7.52   8.32  16.96  20.05 -24.96 -42.69 -47.47  55.04 -57.62   2.03
61.94 64.41 -71.3 93.6 151.65 151.75 -0.43 -3.18 4.59 -5.55
6.44 -9.48 9.31 0.67 -14.34 -8.09 16.23 17.69 19.46 23.52
-52.59]

B = [61.94 16.23 19.46 -5.55 -0.43 93.6]

2 for 循环会做这件事,但我想知道是否有 python 方法可以更快地做到这一点。

我尝试了一个循环,但它不起作用(我怀疑 numpy.where 不适用于不同大小的数组)

    def get_index(self, lst_1, lst_2):
tmp_list = list()
for i in range(min(len(lst_1), len(lst_2))):
if np.where(lst_2[i] == lst_1):
tmp_list.append(i)

return tmp_list

任何建议将不胜感激:)

谢谢

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