If loop within function returns value error when input is an array

Question

I have a function f(x) as follows:

def f(x):
    if x> 3:
       k = x
    else:
       k = 2*x
    return k

f(2)
f([2,3])

If I