Create New Numpy Array Python Code