gpt4 book ai didi

python - 不同大小数组的numpy操作

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

c=np.array([ 0.  ,  0.2,  0.22,  0.89,  0.99])
rnd = np.random.uniform(low=0.00, high=1.00, size=12)

我想看看c中有多少元素小于rnd中的12个随机数。它需要在 numpy 中并且不使用任何列表,这样它会更快。

输出将是一个包含 12 个元素的数组,每个元素描述每个元素有多少个元素小于 rnd 中的相应数字。

最佳答案

您可以使用 broadcastingc1D 扩展到 2D 数组版本后 None/np.newaxis用于以矢量化方式对所有元素进行比较,然后使用 .sum(0) 对行进行求和以进行计数,就像这样 -

(c[:,None] < rnd).sum(0)

看来你也可以使用高效的np.searchsorted像这样 -

np.searchsorted(c,rnd)

关于python - 不同大小数组的numpy操作,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35522087/

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