axis=0:表示沿着行的方向进行操作,即对每一列进行操作。
axis=1:表示沿着列的方向进行操作,即对每一行进行操作。
示例:
import numpy as np# 创建一个二维数组
array = np.array([[1, 2, 3],[4, 5, 6],[7, 8, 9]
])
- axis=0 (表示沿着行的方向对列进行操作)
# 计算每一列的和
column_sum = np.sum(array, axis=0)
/***array: [1, 2, 3],[4, 5, 6],[7, 8, 9] **/
print(column_sum) # 输出: [12 15 18]
即沿着行相加,sum也就等于每一列上所有行相加(即沿着行方向对列进行操作)
- axis=1(表示沿着列的方向对行进行操作)
# 计算每一行的和
row_sum = np.sum(array, axis=1)
/***array: [1, 2, 3],[4, 5, 6],[7, 8, 9] **/
print(row_sum) # 输出: [ 6 15 24]
即沿着列相加,sum也就等于每一行上所有列相加(即沿着列方向对行进行操作)