是否有使用 Python Matplotlib 在分组箱线图上显示样本大小的好方法
Is there a good way to display sample size on grouped boxplots using Python Matplotlib
我可以使用 groupby 获取尺寸信息并将文本添加到相应位置。但我忍不住想有更好的方法,因为这看起来很平凡,很多人都希望看到...
为了说明,以下代码将生成分组箱线图
import pandas as pd
df = pd.DataFrame(rand(100, 1), columns=['value'])
df.ix[:23, 'class']='A'
df.ix[24:, 'class']='B'
df.boxplot(column='value', by='class')
我想要的是显示每个 class A 和 B 的样本大小,即分别为 24 和 76。它可能显示为图例或框附近的某个地方,我都可以。
谢谢!
n 在 class 刻度标签中。我把它当作一个传说来试过,但我认为它并不那么清楚。 R 有更多的箱线图选项,包括使箱的宽度与样本大小成正比;不是 matplotlib 中的默认设置,但简单易读:
import pandas as pd
from numpy.random import rand, randint
df = pd.DataFrame(rand(100, 1), columns=['value'])
cut1 = randint(2,47)
cut2 = randint(52, 97)
df.ix[:cut1, 'class']='A'
df.ix[cut1+1:cut2, 'class']='B'
df.ix[cut2+1:, 'class'] = 'C'
dfg = df.groupby('class')
counts = [len(v) for k, v in dfg]
total = float(sum(counts))
cases = len(counts)
widths = [c/total for c in counts]
cax = df.boxplot(column='value', by='class', widths=widths)
cax.set_xticklabels(['%s\n$n$=%d'%(k, len(v)) for k, v in dfg])
我可以使用 groupby 获取尺寸信息并将文本添加到相应位置。但我忍不住想有更好的方法,因为这看起来很平凡,很多人都希望看到...
为了说明,以下代码将生成分组箱线图
import pandas as pd
df = pd.DataFrame(rand(100, 1), columns=['value'])
df.ix[:23, 'class']='A'
df.ix[24:, 'class']='B'
df.boxplot(column='value', by='class')
谢谢!
n 在 class 刻度标签中。我把它当作一个传说来试过,但我认为它并不那么清楚。 R 有更多的箱线图选项,包括使箱的宽度与样本大小成正比;不是 matplotlib 中的默认设置,但简单易读:
import pandas as pd
from numpy.random import rand, randint
df = pd.DataFrame(rand(100, 1), columns=['value'])
cut1 = randint(2,47)
cut2 = randint(52, 97)
df.ix[:cut1, 'class']='A'
df.ix[cut1+1:cut2, 'class']='B'
df.ix[cut2+1:, 'class'] = 'C'
dfg = df.groupby('class')
counts = [len(v) for k, v in dfg]
total = float(sum(counts))
cases = len(counts)
widths = [c/total for c in counts]
cax = df.boxplot(column='value', by='class', widths=widths)
cax.set_xticklabels(['%s\n$n$=%d'%(k, len(v)) for k, v in dfg])