plt.legend()给图像加上图例，以区分哪个图形属于哪个

Posted 2020-12-21 小小喽啰

用于多个图形画在同一画框中，以区分哪个图形属于哪个

import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0, 2 * np.pi, 50)
y = np.sin(x)
plt.plot(x, y)
plt.plot(x, y * 2)

 

 加上这个函数，就知道它的作用了

import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0, 2 * np.pi, 50)
y = np.sin(x)
plt.plot(x, y,c=\'green\')
plt.plot(x, y * 2,c=\'blue\')
plt.legend([\'green\',\'blue\'])

 

 或者是这样子

import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0, 2 * np.pi, 50)
y = np.sin(x)
plt.plot(x, y, label="sin(x)",c=\'green\')
plt.plot(x, y * 2, label="2sin(x)",c=\'blue\')
plt.legend()