python - Python 中的矩阵大小

Question

a 是一个 2x2 矩阵

b 是一个 2x1 矩阵

c 是一个 1x2 矩阵

但是......d是什么样的矩阵？

import numpy as np
a= np.array([[1,2],[3,4]])
b= np.array([[1],[2]])
c= np.array([[1,2]])
d= np.array([1,2])

Variable explorer

Answer 1

变量d不是矩阵而是行向量。

import numpy as np
a= np.array([[1,2],[3,4]])
b= np.array([[1],[2]])
c= np.array([[1,2]])
d= np.array([1,2])

print(a.shape, b.shape, c.shape, d.shape)
print(a.ndim, b.ndim, c.ndim, d.ndim)

输出形状:

(2, 2) (2, 1) (1, 2) (2,)

和尺寸:

2 2 2 1

括号的数量表示维数，例如:

e = np.array([[[1,2]]])

输出 shape (1, 1, 2) 和 ndim 3 (所以是 3 维)。