matplotlib 如何将误差条添加到交互作用图(statmodels)？

7xllpg7q 于 2023-10-24 发布在其他

关注(0)|答案(2)|浏览(121)

我有以下代码：

import numpy as np
import matplotlib.pyplot as plt
from statsmodels.graphics.factorplots import interaction_plot

a = np.array( [ item for item in [ 'a1', 'a2', 'a3' ] for _ in range(30) ] )
b = np.array( [ item for _ in range(45) for item in [ 'b1', 'b2' ] ] )
np.random.seed(123)
mse = np.ravel( np.column_stack( (np.random.normal(-1, 1, size=45 ), np.random.normal(2, 0.5, size=45 ) )) )
f = interaction_plot( a, b, mse )

其给出：

有没有一种简单的方法可以直接为每个点添加误差线？

f.axes.errorbar()?

还是直接用matplotlib绘制图更好？

matplotlib

来源：https://stackoverflow.com/questions/37908989/how-to-add-error-bars-to-interaction-plot-statsmodels

2条答案

按热度按时间

gijlo24d1#

好吧，看起来这个功能是not yet directly supported，所以我决定直接修改源代码，创建一个新的功能。我把它贴在这里，也许它可以为某人所用。

def int_plot(x, trace, response, func=np.mean, ax=None, plottype='b',
                     xlabel=None, ylabel=None, colors=[], markers=[],
                     linestyles=[], legendloc='best', legendtitle=None,
# - - - My changes !!
                     errorbars=False, errorbartyp='std',
# - - - - 
                     **kwargs):

data = DataFrame(dict(x=x, trace=trace, response=response))
    plot_data = data.groupby(['trace', 'x']).aggregate(func).reset_index()

# - - - My changes !!
    if errorbars:
        if errorbartyp == 'std':
            yerr = data.groupby(['trace', 'x']).aggregate( lambda xx: np.std(xx,ddof=1) ).reset_index()
        elif errorbartyp == 'ci95':
            yerr = data.groupby(['trace', 'x']).aggregate( t_ci ).reset_index()
        else:
            raise ValueError("Type of error bars %s not understood" % errorbartyp)
# - - - - - - -
    n_trace = len(plot_data['trace'].unique())

if plottype == 'both' or plottype == 'b':
        for i, (values, group) in enumerate(plot_data.groupby(['trace'])):
            # trace label
            label = str(group['trace'].values[0])
# - - - My changes !!
            if errorbars:
                ax.errorbar(group['x'], group['response'], 
                            yerr=yerr.loc[ yerr['trace']==values ]['response'].values, 
                        color=colors[i], ecolor='black',
                        marker=markers[i], label='',
                        linestyle=linestyles[i], **kwargs)
# - - - - - - - - - - 
            ax.plot(group['x'], group['response'], color=colors[i],
                    marker=markers[i], label=label,
                    linestyle=linestyles[i], **kwargs)

有了这个，我可以得到这个情节：

f = int_plot( a, b, mse, errorbars=True, errorbartyp='std' )

**注意：**代码也可以使用函数t_ci()来聚合误差条，我是这样定义的：

def t_ci( x, C=0.95 ):
    from scipy.stats import t

    x = np.array( x )
    n = len( x )
    tstat = t.ppf( (1-C)/2, n )
    return np.std( x, ddof=1 ) * tstat / np.sqrt( n )

同样，我只是稍微调整了一下函数以适应我当前的需要。原始函数可以在这里找到：）

赞(0）回复(0）举报 2023-10-24

7bsow1i62#

使用axes.errorbar（）可能是实现误差线的一个不太可靠的方法，但这是我的解决方法。这样做会在图表中添加另一条线，然后您需要将其与交互图的线对齐。

赞(0）回复(0）举报 2023-10-24