我正在使用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()作为函数。我的代码当前生成的数字是:

python-2.7 - 在Python Probplot中更改标记样式/颜色-LMLPHP

最佳答案

关键是与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()

这样创建的图形看起来很难看,但是演示了功能:

python-2.7 - 在Python Probplot中更改标记样式/颜色-LMLPHP

可以从轴上通过更通用的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>]

10-06 02:05