创建带有类别和日期的图片的图

Question

我想知道如何在Python（最好是matplotlib）中进行以下类型的绘图：

我希望沿y轴显示四个类别，然后沿x轴显示日期，如图所示。

我有一个包含两列[类别]，[日期]的CSV文件。日期格式为：dd-mm-yyy。

提取：

category1,05-01-2020

category1,02-02-2020

category3,06-03-2020

category2,12-04-2020

等...

将不胜感激！

Answer 1

您可以按原样绘制类别与日期的关系图。对于颜色代码，您需要将类别转换为单个数字，可以使用pandas Categorical data type轻松实现。

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')