[Numpy-discussion] my code (python version)