[Numpy-discussion] help to speed up the python code