Циклирование строк для построения линий между парами координат в Python (matplotlib)

Question

Я новичок в Python, и я изо всех сил пытаюсь создать следующий сюжет, используя matplotlib:

Со следующим данные:

У меня много строк, поэтому я подумал сделать что-то вроде:

For each row, plot a line from (x_start, y_start) to (x_end, y_end)

Любой совет?

Спасибо!

Answer 1

Вы на правильном пути. Импортируйте данные через что-то вроде pandas.read_csv(). Здесь я просто поместил все в массив, чтобы сделать это простым.

In [1]: import numpy as np

In [2]: import matplotlib.pyplot as plt

In [3]: x = np.array([[1, 1, 3, 1], [2, 2, 4, 2], [3, 3, 5, 3], [4, 4, 6, 4], [5, 5, 7, 5]])

In [4]: plt.figure()
   ...: for row in x:
   ...:     plt.plot([row[0], row[2]], [row[1], row[3]], marker='.')
   ...:

In [5]: plt.show()

Я оставляю надписи на осях и цвета для вас. Подсказка: посмотрите на карты цветов matplotlib.