如何绘制带有x和y分类轴的线图？

Question

我正在尝试绘制比较2个类别变量的折线图。 但是我一直遇到错误。 我将代码放在下面：

import matplotlib.pyplot as plt

cat = ["bored", "happy", "bored", "bored", "happy", "bored"]
dog = ["happy", "happy", "happy", "happy", "bored", "bored"]
activity = ["combing", "drinking", "feeding", "napping", "playing", 
"washing"]

fig, ax = plt.subplots()
ax.plot(activity, dog, label="dog")
ax.plot(activity, cat, label="cat")
ax.legend()

plt.show()

Answer 1

在此处提供答案：当使用matplotlib> = 2.1运行时，来自问题的代码

import matplotlib.pyplot as plt

cat = ["bored", "happy", "bored", "bored", "happy", "bored"]
dog = ["happy", "happy", "happy", "happy", "bored", "bored"]
activity = ["combing", "drinking", "feeding", "napping", "playing", 
"washing"]

fig, ax = plt.subplots()
ax.plot(activity, dog, label="dog")
ax.plot(activity, cat, label="cat")
ax.legend()

plt.show()

运行良好并产生

---

对于较早版本的matplotlib，轴应为数字。 因此，您需要将类别转换为数字，绘制它们，然后相应地设置刻度标签。

 import numpy as np import matplotlib.pyplot as plt cat = ["bored", "happy", "bored", "bored", "happy", "bored"] dog = ["happy", "happy", "happy", "happy", "bored", "bored"] activity = ["combing", "drinking", "feeding", "napping", "playing", "washing"] catu, cati = np.unique(cat, return_inverse=True) dogu, dogi = np.unique(dog, return_inverse=True) fig, ax = plt.subplots() ax.plot(range(len(dog)), dogi, label="dog") ax.plot(range(len(cat)), cati, label="cat") ax.set_xticks(range(len(activity))) ax.set_xticklabels(activity) ax.set_yticks(range(len(catu))) ax.set_yticklabels(catu) ax.legend() plt.show()