python - 如何将字典 kwargs 输入到 matplotlib 图例例程中？

Question

我想编写一个函数，在输出图之前接受图例参数字典。我在下面包含了一个小例子。

导入

import numpy as np
import matplotlib.pyplot as plt

数据

x = np.linspace(0, 100, 501)
y = np.sin(x)

图例参数

legend_dict = dict(ncol=1, loc='best', fancybox=True, shadow=True)
label = 'xy data sample'
# label = None

绘图

if label is not None:
    plt.plot(x, y, label=label, **legend_dict)
else:
    plt.plot(x, y)
plt.show()

这给了我以下错误(可以通过取消注释 label=None 来避免)。

    plt.plot(x, y, label=label, **legend_dict) # this line
AttributeError: Unknown property shadow # this error

为什么这种方法不起作用？

Answer 1

您正在尝试将图例 kwargs 传递给 plot 函数。需要单独调用.legend()

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 100, 501)
y = np.sin(x)

legend_dict = dict(ncol=1, loc='best', fancybox=True, shadow=True)
label = 'xy data sample'
#label = None

plt.plot(x, y, label=label) 
plt.legend(**legend_dict)
plt.show()

注意也不需要 if 语句 - 标签为 None 就可以了，因为这是默认值!