python 注册矩阵
时间: 2023-07-07 07:24:59 浏览: 50
在使用 NumPy 库计算矩阵时,需要先将矩阵注册为 NumPy 数组。可以使用 `array` 函数将列表或元组转换为 NumPy 数组。
例如,假设我们要创建一个 2x2 的矩阵和一个 2x1 的向量,可以使用以下代码:
```python
import numpy as np
# 创建矩阵和向量
A = np.array([[1, 2], [3, 4]])
B = np.array([[5], [6]])
# 打印矩阵和向量
print("A = ")
print(A)
print("B = ")
print(B)
```
输出结果如下:
```
A =
[[1 2]
[3 4]]
B =
[[5]
[6]]
```
在这个例子中,我们使用 `array` 函数将列表 `[[1, 2], [3, 4]]` 和 `[[5], [6]]` 转换为 NumPy 数组,得到矩阵 `A` 和向量 `B`。可以通过打印结果来验证矩阵和向量是否创建成功。
需要注意的是,创建矩阵和向量时,元素的顺序是从左到右、从上到下的。例如,在上面的代码中,矩阵 `A` 的元素顺序为 `1`, `2`, `3`, `4`,向量 `B` 的元素顺序为 `5`, `6`。如果需要改变元素的顺序,可以在创建矩阵和向量时使用列表来指定元素的顺序。例如,如果我们要创建一个 2x2 的矩阵,元素顺序为 `2`, `1`, `4`, `3`,可以使用以下代码:
```python
import numpy as np
# 创建矩阵
A = np.array([[2, 1], [4, 3]])
# 打印矩阵
print("A = ")
print(A)
```
输出结果如下:
```
A =
[[2 1]
[4 3]]
```
在这个例子中,我们使用列表 `[[2, 1], [4, 3]]` 来指定矩阵元素的顺序。