【Python】Numpy array

版权声明：本文为博主原创文章，可以全文或部分转载，但请注明出处。 https://blog.csdn.net/shanshangyouzhiyangM/article/details/85851271

1。计算矩阵的size和dimension
两个函数: array.ndim 和 np.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