python画二维矩阵图_Python:线性代数-矩阵-绘制向量 (五十)

向量 lab

在此 notebook 中，你将学习如何绘制二维向量和进行某些向量计算。

具体来说：

绘制二维向量

将二维向量与标量相乘并绘制结果

将两个向量相加并绘制结果

对于此 lab，我们将使用 python 软件包 NumPy 创建向量并计算向量运算。对于绘图部分，我们将使用 python 软件包 Matplotlib。

绘制二维向量

对于此部分，我们将绘制定义如下的向量 \(\vec{v}\)。

\(\hspace{1cm}\vec{v} = \begin{bmatrix} 1\ 1\end{bmatrix}\)

下面简要介绍了绘制向量 \(\vec{v}\) 的 Python 代码所包含的部分。

使用 import 方法使 NumPy 和 Matlibplot python 软件包可用。

定义向量 \(\vec{v}\)

使用 Matlibplot 绘制向量 \(\vec{v}\)

创建一个变量 ax 来表示图的坐标轴

使用 ax 和 plot 方法绘出原点 0,0（红点）

使用 ax 和 arrow 方法绘出向量 \(\vec{v}\)：蓝色箭头，起点为 0,0

确定 x 轴的格式

使用 xlim 方法设置范围

使用 ax 和 set_xticks 方法设置主要刻度线

确定 y 轴的格式

使用 ylim 方法设置范围

使用 ax 和 set_yticks 方法设置主要刻度线

使用 grid 方法创建网格线

使用 show 方法显示图

# Import NumPy and Matplotlib

%matplotlib inline

import numpy as np

import matplotlib.pyplot as plt

# Define vector v

v = np.array([1,1])

# Plots vector v as blue arrow with red dot at origin (0,0) using Matplotlib

# Creates axes of plot referenced 'ax'

ax = plt.axes()

# Plots red dot at origin (0,0)

ax.plot(0,0,'or')

# Plots vector v as blue arrow starting at origin 0,0

ax.arrow(0, 0, *v, color='b', linewidth=2.0, head_width=0.20, head_length=0.25)

# Sets limit for plot for x-axis

plt.xlim(-2,2)

# Set major ticks for x-axis

major_xticks = np.arange(-2, 3)

ax.set_xticks(major_xticks)

# Sets limit for plot for y-axis

plt.ylim(-1, 2)

# Set major ticks for y-axis

major_yticks = np.arange(-1, 3)

ax.set_yticks(major_yticks)

# Creates gridlines for only major tick marks

plt.grid(b=True, which='major')

# Displays final plot

plt.show()

<