我有以下代码:
import numpy as np
from fitter import Fitter
from matplotlib import pyplot as plt
randomData = np.random.normal(0, 1, 1000)
def plotOfFittedDistributions(data):
f = Fitter(data, distributions=['gamma', "norm", 'invgauss'], timeout=100)
f.fit()
f.hist()
f.plot_pdf()
ax = plt.gcf()
return ax
a = plotOfFittedDistributions(data=randomData)
a
可以看出,我使用
plt.gcf()
访问了当前图形。使用 plt.gcf()
访问后是否可以更改绘图的颜色?所以我想在创建后访问具有多个绘图的单个图形,然后更改它们的颜色。
我找到了答案:
plt.gca().get_lines()[0].set_color("Purple")
plt.gca().get_lines()[1].set_color("Yellow")
plt.gca().get_lines()[2].set_color("Grey")
myLegend = plt.gca().get_legend()
myLegend.legend_handles[0].set_color('Purple')
myLegend.legend_handles[1].set_color('Yellow')
myLegend.legend_handles[2].set_color('Grey')