import numpy as np from expr import f a = np.array( [[1,2], [3,4]]) print f(a)