Создание сюжета в виде картинки с категориями и датами - PullRequest
0 голосов
/ 30 апреля 2020

Мне интересно, как я могу сделать следующий тип сюжета в Python (предпочтительно matplotlib):

enter image description here

Я бы хотел четыре категории вдоль оси Y, а затем даты вдоль оси X, как на рисунке.

У меня есть файл CSV с двумя столбцами [категория], [дата]. Формат даты: дд-мм-гггг.

Извлечение:

категория1,05-01-2020

категория1,02-02-2020

category3,06-03-2020

category2,12-04-2020

et c ...

Помощь будет оценена по достоинству!

1 Ответ

0 голосов
/ 01 мая 2020

Вы можете просто отобразить категории и даты как есть. Для цветового кода вам необходимо преобразовать категории в отдельные числа, что может быть легко достигнуто с помощью pandas Категориальный тип данных .

d = """category1,05-01-2020
category1,02-02-2020
category3,06-03-2020
category2,12-04-2020"""
df = pd.read_csv(StringIO(d), sep=',', parse_dates=[1], header=None, names=['category','date'])


fig, ax = plt.subplots()
ax.scatter(df['date'],df['category'], marker='s', c=df['category'].astype('category').cat.codes, cmap='tab10')

enter image description here

...