具有分裂y轴的matplotlib boxplot

Question

我想用类似于

的数据制作一个箱形图

d = {'Education': [1,1,1,1,2,2,2,2,2,3,3,3,3,4,4,4,4], 
 'Hours absent': [3, 100,5,7,2,128,4,6,7,1,2,118,2,4,136,1,1]}
df = pd.DataFrame(data=d) 
df.head() 

这很有效：

df.boxplot(column=['Hours absent'] , by=['Education'])
plt.ylim(0, 140)
plt.show()

但异常值很远，因此我想分割y轴。 但是这里的boxplot命令＆＃34;列＆＃34; ＆＃34; by＆＃34;不再被接受。因此，我不是通过教育分割数据，而是只获得一个合并的数据点。 这是我的代码：

dfnew = df[['Hours absent', 'Education']] # In reality I take the different 
columns from a much bigger dataset

fig, (ax1, ax2) = plt.subplots(2, 1, sharex=True)

ax1.boxplot(dfnew['Hours absent'])
ax1.set_ylim(40, 140)

ax2.boxplot(dfnew['Hours absent'])
ax2.set_ylim(0, 40)

ax1.spines['bottom'].set_visible(False)
ax2.spines['top'].set_visible(False)

ax1.xaxis.tick_top()
ax1.tick_params(labeltop='off')  # don't put tick labels at the top
ax2.xaxis.tick_bottom()

d = .015  # how big to make the diagonal lines in axes coordinates
# arguments to pass to plot, just so we don't keep repeating them
kwargs = dict(transform=ax1.transAxes, color='k', clip_on=False)
ax1.plot((-d, +d), (-d, +d), **kwargs)        # top-left diagonal
ax1.plot((1 - d, 1 + d), (-d, +d), **kwargs)  # top-right diagonal

kwargs.update(transform=ax2.transAxes)  # switch to the bottom axes
ax2.plot((-d, +d), (1 - d, 1 + d), **kwargs)  # bottom-left diagonal
ax2.plot((1 - d, 1 + d), (1 - d, 1 + d), **kwargs)  # bottom-right diagonal

plt.show()

这些是我尝试的东西（我总是为第一个和第二个子图改变这个）和我得到的错误。

ax1.boxplot(dfnew['Hours absent'],dfnew['Education']) 
#The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), 
#a.any() or a.all().
ax1.boxplot(column=dfnew['Hours absent'], by=dfnew['Education'])#boxplot() 
#got an unexpected keyword argument 'column'
ax1.boxplot(dfnew['Hours absent'], by=dfnew['Education']) #boxplot() got an 
#unexpected keyword argument 'by'

我还尝试将数据转换为y轴的数组，并列出x轴：

data = df[['Hours absent']].as_matrix()
labels= list(df['Education'])

print(labels)
print(len(data))
print(len(labels))

print(type(data))
print(type(labels))

我在剧情命令中代替了这样：

ax1.boxplot(x=data, labels=labels)
ax2.boxplot(x=data, labels=labels)

现在错误是ValueError：标签和X的尺寸必须兼容。 但他们都是17岁，我不明白这里出了什么问题。

Answer 1

您过度复杂，打破Y轴的代码与绘制箱线图的代码无关。没有什么能阻止您使用df.boxplot，它会添加一些您不想要的标签和标题，但这很容易修复。

df.boxplot(column='Hours absent', by='Education', ax=ax1)
ax1.set_xlabel('')
ax1.set_ylim(ymin=90)

df.boxplot(column='Hours absent', by='Education', ax=ax2)
ax2.set_title('')
ax2.set_ylim(ymax=50)
fig.subplots_adjust(top=0.87)

当然，只要您提供所需的参数，您也可以使用matplotlib的boxplot。根据文档字符串，它将使

  

每个x列或每个向量的方框和胡须图   序列x

这意味着你必须自己做“by”部分。

grouper = df.groupby('Education')['Hours absent']
x = [grouper.get_group(k) for k in grouper.groups]

ax1.boxplot(x)
ax1.set_ylim(ymin=90)

ax2.boxplot(x)
ax2.set_ylim(ymax=50)