np.transpose()が図を書いてもたまにわからない。。。