1. 程式人生 > >【Python】Numpy array

【Python】Numpy array

1。計算矩陣的size和dimension
兩個函式: array.ndimnp.size(array, axis)

import numpy as np

# 定義一個二維張量
X=np.array([[1,2,3,4],
              [5,6,7,8],
              [9,10,11,12]])
 
dim=X.ndim # 計算X的維度 = 2
number=X.size  # 計算 X 中所有元素的個數 = 12
X_row=np.size(X,0)  #計算 X 一行元素的個數 = 4
X_col=np.size(X,1)  #計算 X 一列元素的個數 = 3
 
print("number:",number)
print("X_row:",X_row)
print("X_col:",X_col)
 
<<
number: 12
X_row: 3
X_col: 4