Statsmodels plot_sim одновременное сравнение имя_ не работает - PullRequest
Новая
       33

Statsmodels plot_sim одновременное сравнение имя_ не работает

0 голосов
/ 19 сентября 2018

Я использую парное сравнение Тьюкиса в statsmodels, чтобы визуализировать различия между группами.Код работает нормально, когда я запускаю его так: 

from statsmodels.stats.multicomp import pairwise_tukeyhsd
from matplotlib import pyplot as plt 
import numpy as np 
import matplotlib

%matplotlib notebook
fig,ax = plt.subplots()
T = np.asanyarray([1,2,0,1,2,3,4,5,10])
N = np.asanyarray(['a','a','a','b','b','b','c','c','c'])

tk = pairwise_tukeyhsd(T,N)
tk.plot_simultaneous(ax=ax)

Он выводит enter image description here

Однако, когда я пытаюсь использовать сравнение_имя = 'c':

tk.plot_simultaneous(comparison_name ='c',ax=ax)

Сбой, и я получаю следующую трассировку: 

TypeError                                 Traceback (most recent call last)
<ipython-input-10-9d84068180f9> in <module>()
     19 
     20 tk = pairwise_tukeyhsd(T,N)
---> 21 tk.plot_simultaneous(comparison_name ='c',ax=ax)

C:\Anaconda3\lib\site-packages\statsmodels\sandbox\stats\multicomp.py in plot_simultaneous(self, comparison_name, ax, figsize, xlabel, ylabel)
    733                 if self.groupsunique[i] == comparison_name:
    734                     continue
--> 735                 if (min(maxrange[i], maxrange[midx]) -
    736                                          max(minrange[i], minrange[midx]) < 0):
    737                     sigidx.append(i)

TypeError: only integer scalar arrays can be converted to a scalar index

Как я могу это исправить?

...