Python Iterate Numpy Array Transpose