Python / numpy：数组中的所有值都是x？

发布时间：2020-12-20 12:17:55 所属栏目：Python 来源：网络整理

导读：我有一个这样的有序数组：numpy.array([1,2,5,10,25,36,66,90,121,230,333,500]) 假设我希望所有值都高达60(如果60不在,我想停在第一个值大于60),所以我想要[1,66].如果我使用 = 60的numpy.where(),它会在66之前停止. 我的解决方案 from numpy import *x = a

我有一个这样的有序数组：numpy.array([1,2,5,10,25,36,66,90,121,230,333,500])

假设我希望所有值都高达60(如果60不在,我想停在第一个值大于60),所以我想要[1,66].如果我使用< = 60的numpy.where(),它会在66之前停止. 我的解决方案

from numpy import *
x = array([1,500])
print x[:where(x >= 60)[0][0]+1]
>>>[ 1  2  5 10 25 36 66]

解决方法

你不需要任何特殊的numpy.

import numpy,bisect
a = numpy.array([1,500])
idx = bisect.bisect(a,60)
print a[:idx]

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!