已知两点用numpy画直线
时间: 2023-08-08 15:09:49 浏览: 162
可以使用numpy和matplotlib库来绘制直线。假设我们要绘制从点(1,2)到点(4,5)的直线,可以按照以下步骤操作:
1.导入需要的库:
```python
import numpy as np
import matplotlib.pyplot as plt
```
2.定义两个点的坐标:
```python
p1 = np.array([1, 2])
p2 = np.array([4, 5])
```
3.计算直线的斜率和截距:
```python
k = (p2[1]-p1[1])/(p2[0]-p1[0])
b = p1[1] - k*p1[0]
```
4.生成x轴的数据:
```python
x = np.arange(p1[0], p2[0]+1, 0.1)
```
5.计算y轴的数据:
```python
y = k*x + b
```
6.绘制直线:
```python
plt.plot(x, y)
plt.show()
```
完整代码如下:
```python
import numpy as np
import matplotlib.pyplot as plt
# 定义两个点的坐标
p1 = np.array([1, 2])
p2 = np.array([4, 5])
# 计算直线的斜率和截距
k = (p2[1]-p1[1])/(p2[0]-p1[0])
b = p1[1] - k*p1[0]
# 生成x轴的数据
x = np.arange(p1[0], p2[0]+1, 0.1)
# 计算y轴的数据
y = k*x + b
# 绘制直线
plt.plot(x, y)
plt.show()
```
阅读全文