Unpacking variable length list returned from function

后端 未结 5 647
鱼传尺愫
鱼传尺愫 2021-01-25 18:44

Ok so I\'m defining a function that takes a variable number of inputs and clamps each of them

def clamp(*args):
    return [ max(min(arg, 0.8), 0.2) for arg in a         


        
5条回答
  •  渐次进展
    2021-01-25 19:19

    def clamp(*args):
        rt = [max(min(arg, 0.8), 0.2) for arg in args]
        return rt if len(rt) > 1 else rt[0]
    

    or else, remember how you make a tuple of one element. Put a comma after A to force unpack for a single variable.

    A, = clamp(0.12) # with your old function which always returns list
    

提交回复
热议问题