def sqr(x):
    return x * x
def cube(x):
    return x * x * x
f = sqr
print f(2) #Output: 4

a = [sqr, cube]
print a[0](2) #Output: 4
print a[1](2) #Output: 8