Параметрический график в Python с использованием matplotlib
Я пытаюсь построить две функции времени параметрически, используя matplotlib в Python 2.7. Вот что я пытаюсь сделать в своем коде:
import matplotlib.pyplot as plt
import numpy as np
def x(t, x_0, w):
return x_0*np.cos(w*t)
def x_prime(t, x_0, w):
return -x_0*w*np.sin(w*t)
# for x_0 = w = 1:
t_range = np.arange(0, 2*np.pi, np.pi/4)
for t in t_range:
plt.plot(x(t, 1, 1), x_prime(t, 1, 1))
plt.show()
Однако на моем участке ничего не появляется. Оси промаркированы, но ничего не нанесено на график. Раньше я без проблем рисовал внутри циклов for, так почему же это здесь происходит?
Ответы 1
Цикл for - это ваша проблема.
Здесь:
plt.plot(x(t, 1, 1), x_prime(t, 1, 1))
Поскольку t - это одно значение, а не список из-за вашего цикла for, X и y, которые вы пытаетесь построить, являются просто отдельными точками. Отдельные точки не отображаются в matplotlib, если вы не укажете их размер маркера и маркер. Вы можете исправить это, просто нарисовав сразу все точки, чтобы сформировать линию:
def x(t, x_0, w):
print(x)
return x_0*np.cos(w*t)
def x_prime(t, x_0, w):
return -x_0*w*np.sin(w*t)
t_range = np.arange(0, 2*np.pi, np.pi/4)
plt.plot(x(t_range, 1, 1), x_prime(t_range, 1, 1))
plt.show()
Если вместо этого вы по-прежнему хотите использовать метод for-loop для построения точек, вам необходимо добавить дополнительные параметры:
t_range = np.arange(0, 2*np.pi, np.pi/4)
for t in t_range:
print(x(t, 1, 1,), x_prime(t, 1, 1))
plt.plot(x(t, 1, 1), x_prime(t, 1, 1), markersize=3, marker='o')
plt.show()
Обновлено: чтобы все прошло полностью, вы просто снова добавляете первую точку в конец ваших массивов:
a = x_prime(t_range, 1, 1)
b = x(t_range, 1, 1)
a = np.append(a, [a[0]])
b = np.append(b, [b[0]])
plt.plot(a, b)
plt.show()
@WynneT, добавил способ, которым вы можете сделать все наоборот. Если это решение решило ваш вопрос, рассмотрите возможность проголосовать / принять его как решение. Это будет много значить для меня! Спасибо :)
Ах, понятно, спасибо! Сейчас я использую этот метод без цикла for, так как мне нужна линия, соединяющая все точки, и он работает по большей части, но между моей первой и последней точками нет линии. Вы знаете, как я могу заставить линию обойти вокруг, чтобы образовать замкнутую область?