Size Of Numpy Array Python Code