在 matplotlib 上用一个标签标记整个 numpy 数组

Question

我想用一个标签来标记整个 numpy 数组。 例如，以下代码创建 6 (=2+4) 个标签，而不是仅创建 2 个标签：

import numpy as np
import matplotlib.pyplot as plt

a = np.random.rand(10,2)
b = np.random.rand(10,4)


plt.figure()
plt.plot(a, 'blue', label = 'a')
plt.plot(b, 'red', label = 'b')
plt.legend()  

应该如何修改上面的代码以仅创建 2 个图例标签“a”和“b”？

Answer 1

a_lines = plt.plot(a, c='blue')
b_lines = plt.plot(b, c='red')
plt.legend(handles=[a_lines[0], b_lines[0]], labels=['a', 'b'])