python - 在循环迭代中另外比较两个数字

Question

假设有数值变量:a,b,c,d,e...，我想先比较a和b，伪代码:

   if a > b:
       if a > b+c:
           if a > b+c+d:
               ...
           else:
               ...
       else:
           ...
   else:
       ...

一条一条往前走，我不想在代码里写那么多if-else，有什么算法或者简单的方法可以实现吗？

Answer 1

如果您从这些变量中创建一个列表，您可以将 a 与该列表的累积总和进行比较，并找到它为 False 的第一个索引:

import numpy as np
b=1
c=2
d=7
e=10

a=8

x = [b,c,d,e]

np.argmin(np.greater(a,np.cumsum(x)))

试试吧!