for文を使わずに、numpy の割り算でやってみよう - Qiita

ある軸で正規化したいとき 軸毎にスケールが違って、まとめて比較するとつぶれてしまう場合に、軸毎に最大値や合計値で割って正規化して比較したい。 for文でもできるけど、keepdims=True 使うと簡単にできる。 sumした後わざわざ軸追加したけど、keepdimsで同じことができるとは。このためにあるぐらいって気がする。 AAA = np.arange(24).reshape((2,3,4)) print("AAA.shape=", AAA.shape) print("AAA=", AAA) sAAA = AAA.sum(-1, keepdims=True) # sAAA = AAA.sum(-1)[..., np.newaxis] と同じ. nAAA = AAA / sAAA print("\nsAAA.shape=", sAAA.shape) print("\nnAAA=", nA