matplotlibのカラーマップを反転


252

plot_surfaceで使用するために、与えられたカラーマップの色順を単純に逆にする方法を知りたいです。

回答:


463

標準のカラーマップにも、すべて逆バージョンがあります。彼らは_r最後に付け加えられた同じ名前を持っています。(ドキュメントはこちら。


これは「amfhot」では機能しません:「ValueError:カラーマップamfhot_rが認識されません」「hot_r」で十分だと思います。
ショックバーナー、

同様に、「ValueError:カラーマップred_rは認識されません。」
アレックスウィリソン2017年

18

matplotlibでは、カラーマップはリストではありませんが、色のリストがとして含まれていcolormap.colorsます。そして、モジュールmatplotlib.colorsListedColormap()リストからカラーマップを生成する機能を提供します。だからあなたは

colormap_r = ListedColormap(colormap.colors[::-1])

7
+1。ただし、これでカラーマップが全体的に反転することはありません。唯一のListedColormapS(すなわち個別のではなく、補間)を持つcolors属性を。逆転LinearSegmentedColormapsはもう少し複雑です。(_segmentdata辞書のすべてのアイテムを元に戻す必要があります。)
Joe Kington

3
反転についてLinearSegmentedColormapsは、いくつかのカラーマップでこれを実行しました。これに関するIPython Notebookは次のとおりです。
kwinkunks 14

@kwinkunks私はあなたのノートブックの機能が正しくないと思います。以下の回答を参照してください
Mattijn

14

解決策は非常に簡単です。「秋」のカラーマップスキームを使用するとします。標準バージョン:

cmap = matplotlib.cm.autumn

カラーマップのカラースペクトルを反転するには、get_cmap()関数を使用して、次のようにカラーマップタイトルに「_r」を追加します。

cmap_reversed = matplotlib.cm.get_cmap('autumn_r')

.autumnを入手したドキュメントのリンクを提供できますか?
Xitcod13

これは後で壊れる可能性があります... matplotlib.org/3.1.1/gallery/color/colormap_reference.html
Jlanger

13

以下のようにLinearSegmentedColormaps、赤、緑、青の辞書に基づいており、それは各項目を逆にする必要があります。

import matplotlib.pyplot as plt
import matplotlib as mpl
def reverse_colourmap(cmap, name = 'my_cmap_r'):
    """
    In: 
    cmap, name 
    Out:
    my_cmap_r

    Explanation:
    t[0] goes from 0 to 1
    row i:   x  y0  y1 -> t[0] t[1] t[2]
                   /
                  /
    row i+1: x  y0  y1 -> t[n] t[1] t[2]

    so the inverse should do the same:
    row i+1: x  y1  y0 -> 1-t[0] t[2] t[1]
                   /
                  /
    row i:   x  y1  y0 -> 1-t[n] t[2] t[1]
    """        
    reverse = []
    k = []   

    for key in cmap._segmentdata:    
        k.append(key)
        channel = cmap._segmentdata[key]
        data = []

        for t in channel:                    
            data.append((1-t[0],t[2],t[1]))            
        reverse.append(sorted(data))    

    LinearL = dict(zip(k,reverse))
    my_cmap_r = mpl.colors.LinearSegmentedColormap(name, LinearL) 
    return my_cmap_r

それが機能することを確認してください:

my_cmap        
<matplotlib.colors.LinearSegmentedColormap at 0xd5a0518>

my_cmap_r = reverse_colourmap(my_cmap)

fig = plt.figure(figsize=(8, 2))
ax1 = fig.add_axes([0.05, 0.80, 0.9, 0.15])
ax2 = fig.add_axes([0.05, 0.475, 0.9, 0.15])
norm = mpl.colors.Normalize(vmin=0, vmax=1)
cb1 = mpl.colorbar.ColorbarBase(ax1, cmap = my_cmap, norm=norm,orientation='horizontal')
cb2 = mpl.colorbar.ColorbarBase(ax2, cmap = my_cmap_r, norm=norm, orientation='horizontal')

ここに画像の説明を入力してください

編集


user3445587のコメントが届きません。レインボーカラーマップで正常に動作します。

cmap = mpl.cm.jet
cmap_r = reverse_colourmap(cmap)

fig = plt.figure(figsize=(8, 2))
ax1 = fig.add_axes([0.05, 0.80, 0.9, 0.15])
ax2 = fig.add_axes([0.05, 0.475, 0.9, 0.15])
norm = mpl.colors.Normalize(vmin=0, vmax=1)
cb1 = mpl.colorbar.ColorbarBase(ax1, cmap = cmap, norm=norm,orientation='horizontal')
cb2 = mpl.colorbar.ColorbarBase(ax2, cmap = cmap_r, norm=norm, orientation='horizontal')

ここに画像の説明を入力してください

ただし、カスタム宣言されたカラーマップにはデフォルトがない_rため、カスタム宣言されたカラーマップには特に効果的です。次の例は、http//matplotlib.org/examples/pylab_examples/custom_cmap.htmlからの抜粋です。

cdict1 = {'red':   ((0.0, 0.0, 0.0),
                   (0.5, 0.0, 0.1),
                   (1.0, 1.0, 1.0)),

         'green': ((0.0, 0.0, 0.0),
                   (1.0, 0.0, 0.0)),

         'blue':  ((0.0, 0.0, 1.0),
                   (0.5, 0.1, 0.0),
                   (1.0, 0.0, 0.0))
         }

blue_red1 = mpl.colors.LinearSegmentedColormap('BlueRed1', cdict1)
blue_red1_r = reverse_colourmap(blue_red1)

fig = plt.figure(figsize=(8, 2))
ax1 = fig.add_axes([0.05, 0.80, 0.9, 0.15])
ax2 = fig.add_axes([0.05, 0.475, 0.9, 0.15])

norm = mpl.colors.Normalize(vmin=0, vmax=1)
cb1 = mpl.colorbar.ColorbarBase(ax1, cmap = blue_red1, norm=norm,orientation='horizontal')
cb2 = mpl.colorbar.ColorbarBase(ax2, cmap = blue_red1_r, norm=norm, orientation='horizontal')

ここに画像の説明を入力してください


この例は、segmentdataがリストに含まれていないという意味では完全ではないため、必然的に元に戻すことはできません(たとえば、標準のレインボーカラーマップ)。原則として、すべてのLinearSegmentedColormapsは、虹のカラーマップのように、ラムダ関数を使用してプリンシブルにリバーシブルにする必要があると思いますか?
海外

@ user3445587さらにいくつかの例を追加しましたが、標準のレインボーカラーマップでうまく機能すると思います
Mattijn

長すぎるため、新しい答えを追加しました。これは、あらゆる種類のLinearSegmentDataで機能するはずです。問題は、rainbowの場合、_segmentdataの実装が異なることです。したがって、コードは(少なくとも私のマシンでは)、レインボーカラーマップでは機能しません。
海外の

12

Matplotlib 2.0の時点では、およびオブジェクトreversed()用のメソッドがあるため、次のようにすることができますListedColormapLinearSegmentedColorMap

cmap_reversed = cmap.reversed()

ここにドキュメントがあります。


1

LinearSegmentedColormapには2つのタイプがあります。場合によっては、_segmentdataが明示的に指定されます。

>>> cm.jet._segmentdata
{'blue': ((0.0, 0.5, 0.5), (0.11, 1, 1), (0.34, 1, 1), (0.65, 0, 0), (1, 0, 0)), 'red': ((0.0, 0, 0), (0.35, 0, 0), (0.66, 1, 1), (0.89, 1, 1), (1, 0.5, 0.5)), 'green': ((0.0, 0, 0), (0.125, 0, 0), (0.375, 1, 1), (0.64, 1, 1), (0.91, 0, 0), (1, 0, 0))}

rainbowの場合、_segmentdataは次のように与えられます。

>>> cm.rainbow._segmentdata
{'blue': <function <lambda> at 0x7fac32ac2b70>, 'red': <function <lambda> at 0x7fac32ac7840>, 'green': <function <lambda> at 0x7fac32ac2d08>}

matplotlibのソースで関数を見つけることができます。

_rainbow_data = {
        'red': gfunc[33],   # 33: lambda x: np.abs(2 * x - 0.5),
        'green': gfunc[13], # 13: lambda x: np.sin(x * np.pi),
        'blue': gfunc[10],  # 10: lambda x: np.cos(x * np.pi / 2)
}

あなたが望むすべてはすでにmatplotlibで行われています、cm.revcmapを呼び出すだけで、両方のタイプのセグメントデータが反転します。

cm.revcmap(cm.rainbow._segmentdata)

仕事をする必要があります-それから新しいLinearSegmentDataを作成するだけです。revcmapでは、関数ベースのSegmentDataの反転は次のように行われます

def _reverser(f):
    def freversed(x):
        return f(1 - x)
    return freversed

他のリストはいつものように逆になりますが

valnew = [(1.0 - x, y1, y0) for x, y0, y1 in reversed(val)] 

つまり、実際に必要なのは

def reverse_colourmap(cmap, name = 'my_cmap_r'):
     return mpl.colors.LinearSegmentedColormap(name, cm.revcmap(cmap._segmentdata)) 

1

(まだ)任意のカラーマップを元に戻す方法はありませんが、簡単な解決策の1つは、実際にはカラーバーを変更せずに、反転するNormalizeオブジェクトを作成することです。

from matplotlib.colors import Normalize

class InvertedNormalize(Normalize):
    def __call__(self, *args, **kwargs):
        return 1 - super(InvertedNormalize, self).__call__(*args, **kwargs)

次に、これをplot_surface他のMatplotlibプロット関数で使用できます。

inverted_norm = InvertedNormalize(vmin=10, vmax=100)
ax.plot_surface(..., cmap=<your colormap>, norm=inverted_norm)

これは、どのMatplotlibカラーマップでも機能します。


弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.