rank of numpy array in python example