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]) |
コメントを残す