Python3のmatplotlibを使ったヒストグラムの作図

Question

Python3のmatplotlibを使いヒストグラムを作成したいのですが、調べてもうまく作成できないので質問致します。

元となる配列は、
__________price sell buy
0 855000.0 0.2275 0.3814
1 855100.0 4.0390 2.0820
2 855200.0 1.7139 0.5200
3 855300.0 3.7723 1.9440
4 855400.0 3.7017 3.9640
5 855500.0 34.5134 5.8022
6 855600.0 17.2817 2.2116
7 855700.0 8.0095 5.2436
8 855800.0 19.0589 3.3583
9 855900.0 5.8494 6.9031

のようなもので、縦軸をpriceで、横軸は、作成するヒストグラムの左半分がsell、右半分がbuyの列を反映させたいです。

以下を出力するのが目標です。(以下の図中のようにバーの色は変えない)

どなたか詳しい方いらっしゃいましたらご教授頂きたいです。
よろしくお願いします。

Answer 1

matplotlib のラッパーライブラリ seaborn の seaborn.barplot を使って、sell と buy の横方向の棒グラフを2つ作成する。
正の値が右方向 → なので、sell の値は符号を反転させることで左方向 ← にする。
そうすることで、人口ピラミッドのような中央が基点となる左右に伸びた棒グラフが作成できる。

from io import StringIO

import pandas as pd
import seaborn as sns

sns.set(style="whitegrid")

# データフレーム作成
src = StringIO(
    """price         sell         buy
0      855000.0    0.2275   0.3814
1      855100.0    4.0390   2.0820
2      855200.0    1.7139   0.5200
3      855300.0    3.7723   1.9440
4      855400.0    3.7017   3.9640
5      855500.0   34.5134   5.8022
6      855600.0   17.2817   2.2116
7      855700.0    8.0095   5.2436
8      855800.0   19.0589   3.3583
9      855900.0    5.8494   6.9031"""
)
data = pd.read_csv(src, delim_whitespace=True)

# 描画する。
fig, ax = plt.subplots(figsize=(7, 5))

# 棒グラフのデフォルトは右方向→
# sell の値は符号を反転させることで左方向←に棒グラフが作成されるようにする。
copied = data.copy()  # 元のデータを変更しないようにコピーしておく。
copied["sell"] *= -1

colors = ["#60D394", "#FFD97D"]  # 色
names = ["sell", "buy"]  # 列名

for name, color in zip(names, colors):
    sns.barplot(
        x=name,
        y="price",
        data=copied,
        color=color,
        label=name,
        orient="h",
        order=copied["price"].iloc[::-1],
        ax=ax,
    )

ax.set_xlabel("")  # x 軸のラベル
ax.set_ylabel("Price", fontsize=12)  # y 軸のラベル
# x 軸の範囲を左右対称になるように調整する。
max_val = data[["sell", "buy"]].values.max() * 1.1
ax.set_xlim(-max_val, max_val)
ax.legend()  # 凡例追加

plt.show()

追記

次の手順でできます。

1. Axes.get_yticks() で y 軸の目盛りの位置一覧を取り出す。
2. Axes.get_yticklabels() で y 軸の目盛りのラベル一覧を取り出す。
3. Axes.set_yticks() で y 軸の目盛りの位置一覧を10個置きにスライスして設定する。
4. Axes.set_yticklabels() で y 軸の目盛りのラベル一覧を10個置きにスライスして設定する。

from io import StringIO

import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns

sns.set(style="whitegrid")

# データフレーム作成
num_samples = 100
data = pd.DataFrame(
    {
        "price": np.arange(855000, 855000 + num_samples),
        "sell": np.random.uniform(0, 10, num_samples),
        "buy": np.random.uniform(0, 10, num_samples),
    }
)

# 描画する。
fig, ax = plt.subplots(figsize=(7, 5))

# 棒グラフのデフォルトは右方向→
# sell の値は符号を反転させることで左方向←に棒グラフが作成されるようにする。
copied = data.copy()  # 元のデータを変更しないようにコピーしておく。
copied["sell"] *= -1

colors = ["#60D394", "#FFD97D"]  # 色
names = ["sell", "buy"]  # 列名

for name, color in zip(names, colors):
    sns.barplot(
        x=name,
        y="price",
        data=copied,
        color=color,
        label=name,
        orient="h",
        order=copied["price"].iloc[::-1],
        ax=ax,
    )

ax.set_xlabel("")  # x 軸のラベル
ax.set_ylabel("Price", fontsize=12)  # y 軸のラベル
# x 軸の範囲を左右対称になるように調整する。
max_val = data[["sell", "buy"]].values.max() * 1.1
ax.set_xlim(-max_val, max_val)

# y 軸の目盛りの間隔を10個おきに調整する。
yticks = ax.get_yticks()
yticklabels = ax.get_yticklabels()
ax.set_yticks(yticks[::10])
ax.set_yticklabels(yticklabels[::10])

ax.legend()  # 凡例追加

plt.show()

Comments

[yu_2_8_2]

回答ありがとうございます！
おかげさまでうまく実装することができました！

[yu_2_8_2]

すいません、ｙ軸を降順にするのはどうすればよいでしょうか？
ax.yaxis_inverted()
をplt.show()の前に入れても降順になりませんでした・・・

[tiitoi]

boxplot の order 引数で棒グラフの順番 (この場合、price) を指定できます。
そこに、copied["price"].iloc[::-1] で price の値を逆順にして渡せばよいです。
回答のコードを修正しました。

[yu_2_8_2]

ありがとうございます
降順になりました！

[yu_2_8_2]

たびたびすいません。
扱う配列の行が膨大になってくるとそれに対応してy軸の目盛りも増え、文字が潰れてしまうようになってしまいました。
binsを使い間隔を空けようとしましたが、barplotにはないようなのですが、どのように解決したらよいでしょうか？

[tiitoi]

間隔を空ける方法を追記しました。

[yu_2_8_2]

ありがとうございます！
さっそくやってみます！！