np.dotとnp.transposeとif文を使って書き換え