概要
seabornでヒートマップを作る際に、セル内の数値表示が自動的に四捨五入されるらしい。
それによって、相関係数が1になるセルが大量に出てしまうことがあったので、対処した。
基本
以下のようなデータを用意する。
seaborn.py
import seaborn as sns
import matplotlib.pyplot as plt
import pandas as pd
df = pd.DataFrame({'A': range(10),
'B': [x**2 for x in range(10)],
'C': [x**3 for x in range(10)],
'D': [0, 2, 3, 4, 5, 6, 7, 8, 9, 10]})
# A B C D
# 0 0 0 0 0
# 1 1 1 1 2
# 2 2 4 8 3
# 3 3 9 27 4
# 4 4 16 64 5
# 5 5 25 125 6
# 6 6 36 216 7
# 7 7 49 343 8
# 8 8 64 512 9
# 9 9 81 729 10
相関係数は以下のようになる。
seaborn.py
df_corr = df.corr()
print(df_corr)
# A B C D
# A 1.000000 0.962691 0.908434 0.996452
# B 0.962691 1.000000 0.985888 0.944573
# C 0.908434 0.985888 1.000000 0.886333
# D 0.996452 0.944573 0.886333 1.000000
結果は、
seaborn.py
fig, ax = plt.subplots(figsize=(50, 30))
sns.heatmap(df_corr, vmax=1, vmin=-1, center=0, cmap='RdBu', annot=True, annot_kws={"size": 15})
小数点第三位で四捨五入されている。
1. 表示桁数を増やす
セル内の数値表示の桁数を増やすために、オプションにfmt='.3f'を追加。(デフォルト値: ‘.2g’)
seaborn.py
sns.heatmap(df_corr, vmax=1, vmin=-1, center=0, cmap='RdBu', annot=True, annot_kws={"size": 15}, fmt='.3f')
結果は、
小数点第四位で四捨五入されている。
2. 任意の桁で切り捨て
小数点第三位以下を切り捨てたい場合、相関係数の数値を100倍して小数点以下を切り捨てた後、100で割る。
seaborn.py
df_corr = (df.corr() * 100).apply(np.floor) / 100
fig, ax = plt.subplots(figsize=(50, 30))
sns.heatmap(df_corr, vmax=1, vmin=-1, center=0, cmap='RdBu', annot=True, annot_kws={"size": 15}, fmt='.2f')
結果は、
小数点第三位以下が切り捨てられている。