在Python中绘制一组向量(pyplot)

时间:2016-09-20 01:51:42

标签: python arrays numpy

我试图使用pyplot绘制大量的矢量。现在我已经

import matplotlib.pyplot as plt
import numpy as np
import operator

t = np.arange(0, np.pi, .01)


def pos1(t):
    x1 = .72 * np.cos(13*t)
    y1 = .72 * np.sin(13*t)
    return x1, y1


def pos2(t):
    x2 = np.cos(8*t)
    y2 = np.sin(8*t)
    return x2, y2


def vec(t):
    x1 = pos1(t)[0]
    x2 = pos2(t)[0]
    y1 = pos1(t)[1]
    y2 = pos2(t)[1]
    x = np.subtract(x1, x2)
    y = np.subtract(y1, y2)
    return x, y


X = pos2(t)[0]
Y = pos2(t)[1]
U = vec(t)[0]
V = vec(t)[1]

plot1 = plt.figure()
plt.quiver(X, Y, U, V, headlength=4)
plt.show(plot1)

其中pos1(t)pos2(t)vec(t)是返回([a,...,z],[a1,...,z1])形式元组的函数。

这个情节给了我一些接近我想要的东西,但矢量长度都是错误的。两个函数pos1(t),pos2(t)返回特定曲线上的点的元组,vec(t)函数是它们的差异,导致从第一条曲线上的点到第二条曲线上的点的向量。我的情节有正确的方向,但幅度不大。

enter image description here

1 个答案:

答案 0 :(得分:1)

quiver处理箭头的长度。似乎quiver不是你需要的。

使用常规plot

import numpy as np
import matplotlib.pyplot as plt


t = np.arange(0, 2 * np.pi, 0.01)
x0 = np.sin(8 * t)
y0 = np.cos(8 * t)
x1 = 0.72 * np.sin(13 * t)
y1 = 0.72 * np.cos(13 * t)

data = np.column_stack((x0, x1, y0, y1)).reshape(-1, 2)

plt.plot(*data, color='black')
plt.gca().set_aspect('equal', adjustable='box')
plt.show()

<强>结果:

Result原件:

Original