python for if one liner

Question

我有一个问题。

我想让我的代码简单易懂。

我正在尝试从数据 x 中获取下一个日期的值。

这是我的代码。有没有办法使用 lambda 或 map 来缩短它？

def nextDay(date,x,time=1):
     res, c = None, 0

     while c<time:
          temp = iter(x)
          for key in temp:
               if key == date:
                    res = next(temp, None)
          date = res
          c+=1
     return res

x = {'2020-01-11': 3.4, '2020-01-13': 4.1, '2020-02-02': 4.1 }
print(x[nextDay('2020-01-11', x, time=1)])

输出:

4.1

Answer 1

与其遍历日期 time 次并且总是只前进一个日期，不如搜索给定日期然后读取下一个 time 日期。

def nextDay(date, x, time=1):
    it = iter(x)
    date in it
    for _ in range(time):
        date = next(it, None)
    return date

或用itertools.islice :

def nextDay(date, x, time=1):
    it = iter(x)
    date in it
    return next(islice(it, time - 1, None), None)