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维）。

Answer 2

这是一维数组，其中包含2个元素。

检查sandbox中的输出。