如何获得python中提供的值的下一个最小值?它有内置功能吗?
>>>num_list=[1,2,3,4]
>>> min(num_list)
1
>>> max(num_list)
4
如何找到下一个最低到3或下一个到2?预期结果为2和3。
答案 0 :(得分:4)
TL; DR min(n for n in my_list if n>lower_bound)
或max(n for n in my_list if n<upper_bound)
找到立即最低或立即最大的更快的替代方法是numpy
>>> import numpy as np
>>> np.random.seed(10)
>>> a = np.random.random(10000)
>>> a[a>0.7].min()
0.69999533217645671
>>> a[a<0.7].max()
0.70003449227846715
如果您使用numpy
机器感到不舒服
并希望简单地处理一个列表
>>> a = list(a)
然后您可以使用min
和max
内置以及生成器
表达式
>>> min(n for n in a if n>0.7)
0.69999533217645671
>>> max(n for n in a if n<0.7)
0.70003449227846715
>>>
当然,使用列表会得到相同的结果,但要注意性能存在差异:使用ipython
和%timeit
来获取时序,使用{{1}时我有871μs和13.8毫秒使用常规列表的100000元素数组/前面例子的列表。
HTH,ciao
Post Scriptum
我的答案中的解决方案都是O(n),与使用排序的方法的O(n log n)进行比较 - 此外,对于大型数据集,numpy
方法应该< / em>(斜体,因为我手边没有测试......)会受到一个小的乘法因素的影响。
答案 1 :(得分:3)
我看到你的问题被标记为[下限]和[上限]。如果列表已排序,则Python具有等效的C ++&lt; algorithm&gt;的lower_bound
和upper_bound
。他们在bisect
模块中。它们返回开始的索引,并在某个特定值的范围结束后立即返回。
In [1]: import bisect
In [2]: A = [0, 1, 3, 3, 5]
In [3]: A[bisect.bisect_left(A, 3)-1]
Out[3]: 1
In [4]: A[bisect.bisect_right(A, 3)]
Out[4]: 5
答案 2 :(得分:2)
下一个最低到3:
max([x for x in num_list if x < 3])
下一个最大的2:
min([x for x in num_list if x > 2])
答案 3 :(得分:1)
您可以使用sorted
:
>>> l=sorted(num_list,reverse=True)
>>> l[l.index(3)+1]
2
但是,正如Frerich Raabe在评论中所说的那样,你不需要痛苦的整个列表,你可以找到低于3的元素的最大值:
>>> max(i for i in num_list if i<3)
2
对于2之后的最大游戏,您可以使用min
:
>>> min(i for i in num_list if i>2)
3
答案 4 :(得分:1)
使用heapq.nlargest
和heapq.nsmallest
import heapq
num_list = [1, 2, 3, 4]
heapq.nlargest(2, num_list)
heapq.nsmallest(2, num_list)
#>>> [4, 3]
#>>> [1, 2]
答案 5 :(得分:0)
您可以使用以下方法:
num_list = [1,2,3,4]
inds = sorted(range(len(num_list)), key=lambda k: num_list[k])
然后,inds [1]将包含下一个最低元素的索引,依此类推。 此外,您可以使用以下代码而不进行排序:
minv = min(num_list)
nmin = min(nm for nm in num_list if nm > minv)
maxv = max(num_list)
nmax = max(nm for nm in num_list if nm < maxv)
答案 6 :(得分:0)
提供的答案很好,但如果我可以提出建议 - 如果有时可以重复这些值,例如
num_list = [2, 2, 4, 4, 6, 7, 8, 9]
......等等,只是对列表进行排序并获得第一个索引可能不是您想要的。
首先通过set()
传递,您将确保每个条目都是单身:
def sorted_ordered_list(sequence):
return sorted(list(set(sequence)))
然后,您可以将返回的list
索引为您要查找的任何值,从索引0的最低值到最高值。
示例:
>>> my_list = [1, 5, 4, 3, 6, 3, 8, 3, 6, 7, 4, 2, 6, 7, 9, 8, 8]
>>> sorted_ordered_list(my_list)
[1, 2, 3, 4, 5, 6, 7, 8, 9] # now index the list for the desired value
>>>