多列的大熊猫盒子图

时间:2016-06-30 10:38:12

标签: python pandas boxplot

我的数据框架(pandas' s结构)如上所示 enter image description here

现在我想在单独的画布上为每个功能制作boxplot。分离条件是第一列。我有类似的直方图图(下面的代码),但我不能为boxplot制作工作版。

 hist_params = {'normed': True, 'bins': 60, 'alpha': 0.4}
# create the figure
fig = plt.figure(figsize=(16,  25))
for n, feature in enumerate(features):
    # add sub plot on our figure
    ax = fig.add_subplot(features.shape[1] // 5 + 1, 6, n + 1)
    # define range for histograms by cutting 1% of data from both ends
    min_value, max_value = numpy.percentile(data[feature], [1, 99])
    ax.hist(data.ix[data.is_true_seed.values == 0, feature].values, range=(min_value, max_value), 
             label='ghost', **hist_params)
    ax.hist(data.ix[data.is_true_seed.values == 1, feature].values, range=(min_value, max_value), 
             label='true', **hist_params)
    ax.legend(loc='best')

    ax.set_title(feature)

以上代码产生的输出为(仅附加部分):  enter image description here

1 个答案:

答案 0 :(得分:7)

DataFrame.boxplot()可以很好地自动化:

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

df = pd.DataFrame({'is_true_seed': np.random.choice([True, False], 10),
                   'col1': np.random.normal(size=10),
                   'col2': np.random.normal(size=10),
                   'col3': np.random.normal(size=10)})

fig, ax = plt.subplots(figsize=(10,  10))
df.boxplot(['col1', 'col2', 'col3'], 'is_true_seed', ax)

第一个参数告诉pandas要绘制的列,第二个要分组的列(你称之为分离条件),以及第三个要绘制轴的列。

列出所有列,但要分组的列可能会变得乏味,但您可以通过省略第一个参数来避免它。然后,您必须明确命名另外两个:

df.boxplot(by='is_true_seed', ax=ax)