Daily Temperatures

class Solution:
    def dailyTemperatures(self, temperatures):
        stk = []
        cache = {}
        for tx, temp in enumerate(temperatures):
            while stk and stk[-1][0] < temp:
                _, px = stk.pop()
                cache[px] = tx - px
            stk.append([temp, tx])
        
        return list(map(lambda tup: cache.get(tup[0], 0), enumerate(temperatures)))