Python axis意思

在Python中,"axis"這個詞通常用於NumPy(Numerical Python)庫中,這是一個用於科學計算的基礎包。NumPy中的軸(axis)是一個數學概念,指的是數組或者矩陣中的維度。

當你對一個數組或矩陣進行操作時,你可以沿著特定的軸(axis)進行操作。軸的編號從0開始,即最外層維度為軸0,次外層維度為軸1,依此類推。例如:

你可以使用NumPy的ndarray.ndim屬性來獲取數組或矩陣的維度(軸的數量),使用ndarray.shape屬性來獲取數組的維度大小,使用ndarray.ndarray.reshape方法來改變數組的形狀,以及使用ndarray.ndarray.transpose方法來轉置數組(交換軸)。

在NumPy中,你可以使用ndarray.ndarray.sum()ndarray.ndarray.mean()ndarray.ndarray.max()ndarray.ndarray.min()等方法來沿著特定的軸(axis)進行操作。例如,如果你想計算一個二維數組的所有元素的總和,你可以使用sum()方法並指定軸參數,如下所示:

import numpy as np

# 創建一個二維數組(矩陣)
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 計算沿著軸0(行)的總和
row_sums = np.sum(matrix, axis=0)
print(row_sums)  # 輸出: [10 15 20]

# 計算沿著軸1(列)的總和
column_sums = np.sum(matrix, axis=1)
print(column_sums)  # 輸出: [ 6 15 24]

在這個例子中,axis=0表示沿著行的方向進行求和,而axis=1表示沿著列的方向進行求和。