我正在尝试使用字典来更改插入到函数参数中的值。
horizon = 'YEARS'
Dict1 = {'YEARS':'weeks = 52*x','WEEKS':'weeks = x','DAYS':'days = x','HOURS': 'hours = x'}
dummy = pd.to_datetime(pd.datetime.now())
dummy2 = [dummy - timedelta(weeks=x) for x in range(1,5)]
dummy3 = [dummy - timedelta((Dict1[horizon])) for x in range(1,5)]
在这种情况下,dummy2
变量有效,但dummy3
变量却不如Dict1[horizon][0]
给出字符串。
我可以通过其他任何方式实现这一点吗?
答案 0 :(得分:0)
是的,可以利用字典解压缩和lambda函数。
Dict1 = {'YEARS': lambda a: {'weeks': 52 * a}, ...
... timedelta(**Dict1[horizon](x)) ...
这是如何工作的?
好吧,首先Dict1['YEARS']
返回一个函数:lambda a: {'weeks': 52 * a}
。例如,当用2
调用时,它将返回一个元素字典:{'weeks':104}。然后,当传递给函数时,它会由双星(**)拆包,将timedelta(**{'weeks': 104})
变成timedelta(weeks=104)
。正是我们想要的。