[Numpy-discussion] can i improve this code?