使用Numpy模块操作矩阵

前言

尝试根据b站大佬来扩充自己的知识面

使用Numpy模块来实现对矩阵的基本运算以及矩阵元素的基本查找

参考

B站视频

过程

代码片段

import numpy as np
data1 = np.array([1, 2, 3, 4, 5])
data2 = np.array([[1, 2], [3, 4]])
print(data1)    # 打印矩阵内容
print(data2)    # 打印矩阵内容

print(data1.shape, data2.shape)  # 打印矩阵维度

print(np.zeros([2, 3]))     # 打印一个两行三列的空矩阵
print(np.ones([2, 2]))      # 打印一个两行两列的1矩阵

# 实现矩阵的数据更改
data2[1, 0] = 5
print(data2)

# 实现矩阵数据的查找
print(data2[1, 1])  # 或者是下面这个
print(data2[1][1])

# 矩阵的基本运算
data3 = np.ones([2, 3])
print(data3*2)
print(data3/3)
print(data3+3)
print(data3-3)


# 矩阵的加法乘法
data4 = np.array([[1, 2, 3], [4, 5, 6]])
print(data3+data4)  # 对应元素相加
print(data3 * data4)

运行结果如下：

[1 2 3 4 5]
[[1 2]
 [3 4]]
(5,) (2, 2)
[[0. 0. 0.]
 [0. 0. 0.]]
[[1. 1.]
 [1. 1.]]
[[1 2]
 [5 4]]
4
4
[[2. 2. 2.]
 [2. 2. 2.]]
[[0.33333333 0.33333333 0.33333333]
 [0.33333333 0.33333333 0.33333333]]
[[4. 4. 4.]
 [4. 4. 4.]]
[[-2. -2. -2.]
 [-2. -2. -2.]]
[[2. 3. 4.]
 [5. 6. 7.]]
[[1. 2. 3.]
 [4. 5. 6.]]