ホームページ  >  記事  >  バックエンド開発  >  [Python チュートリアル] ウォーターフォール チャートの描画

[Python チュートリアル] ウォーターフォール チャートの描画

黄舟
黄舟オリジナル
2017-02-07 16:09:212768ブラウズ

ウォーターフォールチャートはマッキンゼーコンサルティング社が独自に作成したチャートタイプで、滝のように見えることからウォーターフォールプロットと呼ばれています。この種類のグラフは、絶対値と相対値の組み合わせを使用し、複数の特定の値間の定量的な関係を表現するのに最も適しています。この記事では、Python を使用してこのグラフを描画する方法を簡単に紹介します。

コマンドは次のとおりです

1)导入程序包
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from matplotlib.ticker import FuncFormatter
2)导入及清理数据
def money(x, pos):
return "${:,.0f}".format(x)
formatter = FuncFormatter(money)
index = ['sales','returns','credit fees','rebates','late charges','shipping']
data = {'amount': [350000,-30000,-7500,-25000,95000,-7000]}
trans = pd.DataFrame(data=data,index=index)
blank = trans.amount.cumsum().shift(1).fillna(0)
total = trans.sum().amount
trans.loc["net"]= total
blank.loc["net"] = total
step = blank.reset_index(drop=True).repeat(3).shift(-1)
step[1::3] = np.nan
blank.loc["net"] = 0
3)绘制图像
my_plot = trans.plot(kind='bar', stacked=True, bottom=blank,legend=None, figsize=(10, 5), title="2014 Sales Waterfall")
my_plot.plot(step.index, step.values,'k')
my_plot.set_xlabel("Transaction Types")
my_plot.yaxis.set_major_formatter(formatter)
y_height = trans.amount.cumsum().shift(1).fillna(0)
max = trans.max()
neg_offset = max / 25
pos_offset = max / 50
plot_offset = int(max / 15)
loop = 0
for index, row in trans.iterrows():
if row['amount'] == total:
y = y_height[loop]
else:
y = y_height[loop] + row['amount']
if row['amount'] > 0:
y += pos_offset
else:
y -= neg_offset
my_plot.annotate("{:,.0f}".format(row['amount']),(loop,y),ha="center")
loop+=1
my_plot.set_ylim(0,blank.max()+int(plot_offset))
my_plot.set_xticklabels(trans.index,rotation=0)
my_plot.get_figure().savefig("waterfall.png",dpi=200,bbox_inches='tight')

出力は次のとおりです

[Python チュートリアル] ウォーターフォール チャートの描画

上記は、ウォーターフォールチャートを描画するための[Pythonチュートリアル]の内容です。その他の関連コンテンツについては、PHPの中国語Webサイト( www.php.cn)!


声明:
この記事の内容はネチズンが自主的に寄稿したものであり、著作権は原著者に帰属します。このサイトは、それに相当する法的責任を負いません。盗作または侵害の疑いのあるコンテンツを見つけた場合は、admin@php.cn までご連絡ください。