我正在使用Python 2.7,并已使用scipy.stats.probplot函数创建了概率图。我想更改图形的元素,例如标记的颜色/形状/大小以及最适合趋势线的颜色/宽度。 probplot的文档似乎没有更改这些项目的任何选项。
这是我的代码(相关部分):
#data is a list of y-values sampled from a lognormal distribution
d = getattr(stats, 'lognorm')
param = d.fit(data)
fig = plt.figure(figsize=[6, 6], dpi=100)
ax = fig.add_subplot(111)
fig = stats.probplot(data, dist='lognorm', sparams=param, plot=plt, fit=False)
#These next 3 lines just demonstrate that some plot features
#can be changed independent of the probplot function.
ax.set_title("")
ax.set_xlabel("Quantiles", fontsize=20, fontweight='bold')
ax.set_ylabel("Ordered Values", fontsize=20, fontweight='bold')
plt.show()
我尝试获取xy数据并使用ax.get_xydata()和fig.get_xydata()创建自己的散点图。但是,这两个都失败了,因为两个对象都不具有get_xydata()作为函数。我的代码当前生成的数字是:
最佳答案
关键是与matplotlib结合使用。
您可以使用ax.get_lines()
从line object 访问axes object。然后,可以相应地更改属性。
您可能必须弄清楚哪个索引与标记相关,哪个索引与线相关。在下面的示例中,标记首先出现,因此:
ax.get_lines()[0].set_marker('p')
趋势线是第二:
ax.get_lines()[1].set_linewidth(12.0)
以下示例基于probplot documentation:
import numpy as np
from scipy import stats
import matplotlib.pyplot as plt
nsample = 100
np.random.seed(7654321)
fig = plt.figure()
ax = fig.add_subplot(111)
x = stats.t.rvs(3, size=nsample)
res = stats.probplot(x, plot=plt)
ax.get_lines()[0].set_marker('p')
ax.get_lines()[0].set_markerfacecolor('r')
ax.get_lines()[0].set_markersize(12.0)
ax.get_lines()[1].set_linewidth(12.0)
plt.show()
这样创建的图形看起来很难看,但是演示了功能:
可以从轴上通过更通用的
r^2=0.9616
访问文本(get_children
):ax.get_children()[2].set_fontsize(22.0)
如果没有这些项目的索引的详细知识,则可以尝试:
print ax.get_children()
这给你:
[<matplotlib.lines.Line2D object at 0x33f4350>, <matplotlib.lines.Line2D object at 0x33f4410>,
<matplotlib.text.Text object at 0x33f4bd0>, <matplotlib.spines.Spine object at 0x2f2ead0>,
<matplotlib.spines.Spine object at 0x2f2e8d0>, <matplotlib.spines.Spine object at 0x2f2e9d0>,
<matplotlib.spines.Spine object at 0x2f2e7d0>, <matplotlib.axis.XAxis object at 0x2f2eb90>,
<matplotlib.axis.YAxis object at 0x2f37690>, <matplotlib.text.Text object at 0x2f45290>,
<matplotlib.text.Text object at 0x2f45310>, <matplotlib.text.Text object at 0x2f45390>,
<matplotlib.patches.Rectangle object at 0x2f453d0>]