这个函数该如何编写啊……求解答

def foo(x):
if x%2==0:
return 2*x
else:
return x**3
res = [foo(x) for x in [5,6,7,8]]
print(res)