Lazy evaluation of single assigned numpy expression