重构 Python 代码

Question

任何人都可以就将以下代码重构为 1-2 行以使其简洁易读提出一些建议。

def func(a, b):
    c = []
    if a:
        c.append(a)
    if b:
        c.append(b)
    return c

我试过以下方法：

def func(a, b):
    return [a*(a is not None), b*(b is not None)]

如果a或者b是一个字典等就不行

Answer 1

def func(a, b):
    return [x for x in [a, b] if x]