python - Python 列表的 SQL TOP 1 模拟

Question

这是我输入的 csv 文件的示例:

...
0.7,0.5,0.35,14.4,0.521838919218

0.7,0.5,0.35,14.4,0.521893472678

0.7,0.5,0.35,14.4,0.521948026139

0.7,0.5,0.35,14.4,0.522002579599
...

我需要选择最后一个 float > 随机数所在的顶行。我当前的实现非常慢(脚本有很多此循环和外部循环的迭代):

for line in foo:
   if float(line[-1]) > random.random():
      res = line
      break
...

我怎样才能让它变得更好更快？

编辑:

有人建议我使用 bisect 来完成这项任务，但我不知道该怎么做。

Answer 1

最快的方法是使用二分法(假设 float 列表是有序的)。你可以这样做:

import bisect

float_list = [line[-1] for line in foo]
index = bisect.bisect(float_list, random.random())
if index < len(float_list)
    result = foo[index]
else:
    result = None # None exists

必须对 float 列表进行排序才能使其工作。