Python : matplotlibのヒストグラムで"n未満" , "n以上" で描画する方法

Question

前提・実現したいこと

PythonのMatplotlibでhist()を用いて度数分布 , ヒストグラムを作成したいと思っています。

import os 
import re
import glob
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import japanize_matplotlib 

Age_List = [23, 22, 23, 22, 24, 20, 22, 24,
29, 28, 25, 25, 26, 27, 28, 27, 25, 25, 27, 25, 25,
32, 32, 32, 33, 33, 32, 33, 32, 30, 33, 32,
35, 39, 38, 38, 37, 35, 38, 35, 35, 38, 35, 37,
43, 44, 40, 41, 44, 41, 40, 43, 44, 41, 41, 44, 43, 42, 40, 44, 42, 41, 42,
47, 49, 49, 46, 48, 45, 49, 49, 49, 49, 49, 48, 46, 49, 45, 48, 49, 48,
54, 52, 54, 53, 53, 54, 50, 51, 52, 54,
58, 56, 58, 58, 55, 57, 56, 56, 55]

fig = plt.figure(figsize=(5.5,3.5))
ax = fig.add_subplot(1,1,1)

n, bins, _ = ax.hist(Age_List, bins=10, range=(30, 50),rwidth=0.8)


xs = (bins[:-1] + bins[1:])/2
ys = n.astype(int)

for x, y in zip(xs, ys):
    plt.text(x, y, str(y), horizontalalignment="center")

ヒストグラムを作成するにあたり、Age_Listの値を基に30代未満 , 3032歳 , 3234歳...48~50歳,50以上
のような度数分布で描画したいです。

しかし、今の段階だと30歳未満 , 50歳以上の分布が表示されず除外されてしまいます。

Age_Listの値を基に集計を行えって、度数分布表を作成すれば恐らく作れるとは思うのですがmatplotlib上で簡単に作成する方法があれば教えて欲しいです。

よろしくお願いいたします。

Answer 1

こういうので、どうでしょう。

import matplotlib.pyplot as plt

Age_List = [23, 22, 23, 22, 24, 20, 22, 24,
29, 28, 25, 25, 26, 27, 28, 27, 25, 25, 27, 25, 25,
32, 32, 32, 33, 33, 32, 33, 32, 30, 33, 32,
35, 39, 38, 38, 37, 35, 38, 35, 35, 38, 35, 37,
43, 44, 40, 41, 44, 41, 40, 43, 44, 41, 41, 44, 43, 42, 40, 44, 42, 41, 42,
47, 49, 49, 46, 48, 45, 49, 49, 49, 49, 49, 48, 46, 49, 45, 48, 49, 48,
54, 52, 54, 53, 53, 54, 50, 51, 52, 54,
58, 56, 58, 58, 55, 57, 56, 56, 55]

Age_List_m = [28 if x < 30 else 52 if x > 50 else x for x in Age_List]

fig = plt.figure(figsize=(5.5,3.5))
ax = fig.add_subplot(1,1,1)
n, bins, _ = ax.hist(Age_List_m, bins=13, range=(27, 53),rwidth=0.8)
xs = (bins[:-1] + bins[1:])/2
ys = n.astype(int)
for x, y in zip(xs, ys):
    plt.text(x, y, str(y), horizontalalignment="center")

plt.show()

実行結果

Comments

[Savanof]

ありがとうございます。

ワンライナーだと上手く解釈できなかったのですが、多分この様な事ですよね?

for i in range(len(Age_List)):

num = Age_List[i]

if num < 30:
Age_List[i] = 28
elif i > 50:
Age_List[i] = 52

シンプルに書けて気に入ってます。ありがとうございます。

Answer 2

以下のようにbinsを指定してみてはいかがでしょうか。
詳細はmatplotlib.pyplot.histを参照ください。

import matplotlib.pyplot as plt

Age_List = [23, 22, 23, 22, 24, 20, 22, 24,
29, 28, 25, 25, 26, 27, 28, 27, 25, 25, 27, 25, 25,
32, 32, 32, 33, 33, 32, 33, 32, 30, 33, 32,
35, 39, 38, 38, 37, 35, 38, 35, 35, 38, 35, 37,
43, 44, 40, 41, 44, 41, 40, 43, 44, 41, 41, 44, 43, 42, 40, 44, 42, 41, 42,
47, 49, 49, 46, 48, 45, 49, 49, 49, 49, 49, 48, 46, 49, 45, 48, 49, 48,
54, 52, 54, 53, 53, 54, 50, 51, 52, 54,
58, 56, 58, 58, 55, 57, 56, 56, 55]

fig = plt.figure(figsize=(5.5,3.5))
ax = fig.add_subplot(1,1,1)

bins = [0] + list(range(30,51,2)) + [max(Age_List) + 1]
print(bins) # [0, 30, 32, 34, 36, 38, 40, 42, 44, 46, 48, 50, 59]
n, bins, _ = ax.hist(Age_List, bins=bins, rwidth=0.8)


xs = (bins[:-1] + bins[1:])/2
ys = n.astype(int)

for x, y in zip(xs, ys):
    plt.text(x, y, str(y), horizontalalignment="center")

plt.show()

Comments

[Savanof]

回答ありがとうございます。

公式ドキュメントを読んでみて、

>[1, 2, 3, 4]
then the first bin is [1, 2) (including 1, but excluding 2) and the second [2, 3). The last bin, however, is [3, 4], which includes 4.

この部分の記述を適用したのだと思うのですが、今回の

bins = [0] + list(range(30,51,2)) + [max(Age_List) + 1]

だと、[0] と [max(Age_List) + 1]が極端に棒の幅が大きくなってしまうのですが何か対処方法 or 参考になりそうなドキュメント等などありますでしょうか?

よろしくお願いいたします。