gpt4 book ai didi

python - 由其他数组索引 numpy 数组作为索引

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

我正在尝试获取数组

a = [1,5,4,5,7,8,9,8,4,13,43,42]

和数组

b = [3,5,6,2,7]

我希望 b 成为 a 中的索引,例如一个新数组是

[a[b[0]], a[b[1]], a[b[2]], a[b[3]] ...]

所以 b 中的值是 a 的索引。a 中有 500k 个条目,b 中有 500k 个条目(大约)。有没有一种快速的方法可以启动 numpy 中的所有内核来做到这一点?我已经在 for 循环中做得很好了,它是 sloooooooowwwwww。

编辑澄清。该解决方案必须适用于 2D 和 3D 阵列。所以也许

b = [(2,3), (5,4), (1,2), (1,0)]

我们想要

c = [a[b[0], a[b[1], ...]

最佳答案

并不是说它很快,但 numpy 的方式很简单:

a[b]

输出:

array([5, 8, 9, 4, 8])

关于python - 由其他数组索引 numpy 数组作为索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57845633/

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