axis=1 は横で行 axis=0は縦で列
ベクトルは計算後の数値を出力していく順番
## 2×4の行列を作成 | |
A = np.arange(8).reshape(2,4) | |
# >>> A | |
# array([[0, 1, 2, 3], | |
# [4, 5, 6, 7]]) | |
## 列方向の和をとり,行ベクトルを出力 | |
row_sum = np.sum(A, axis=0) | |
# >>> row_sum | |
# array([ 4, 6, 8, 10]) | |
## 行方向の和をとり,列ベクトルを出力 | |
col_sum = np.sum(A, axis=1) | |
# >>> col_sum | |
# array([ 6, 22]) |