matplotlibで描く円グラフについて

Question

###前提・実現したいこと
matplotlibの持つ円グラフ描画メソッドpieで任意のカラーバリエーションで円グラフが描画出来るようにしたい

###発生している問題・エラーメッセージ
colormapを使用しようと試みました

plt.pie(num, labels=data, colors=plt.cm.get_cmap('RdBu'))

実行結果

$ TypeError: object of type 'LinearSegmentedColormap' has no len()

###補足情報
colorsを指定することはできますし、可変長のデータに対して色を設定することもデータの長さを取得してその分を細分化すれば良いとは思うのですが、colormapの適用方法があればそちらの方がスマートなので、もし適用する方法があれば教えていただきたいです

Answer 1

実現したい事はこーいう事でしょうか？

# -*- coding: utf-8 -*-
import matplotlib.pyplot as plt
from matplotlib import cm, colors

def get_colormap(num):
    normalize = colors.Normalize(vmin=min(num), vmax=max(num))
    color_map = cm.get_cmap('RdBu')
    #color_map = cm.get_cmap('PuOr')
    return color_map(normalize(num))

def main():
    num = range(12)
    data = range(12)
    plt.pie(num, labels=data, colors=get_colormap(num))
    plt.show()

if __name__ == '__main__':
    main()

□参考
matplotlib.colors.Normalize
http://matplotlib.org/api/colors_api.html#matplotlib.colors.Normalize

Comments

[Yusei]

流れとしてはそのような形です。ありがとうございます
plotなどではcolormapが引数で指定でき、pieにも同様の方法がないか気になったため質問しました
ただ、回答してくださったコードの量であればさほど差異があるものでもないので今後はこのコードを基準に実装を考えて行きたいと思います
ありがとうございました