Numpy Array Row Column In Python A Function