So ändern Sie die Farbe, wenn Sie mit Python Seaborn eine Heatmap (Heatmap) zeichnen

Ich habe gesehen, dass einige neuere Artikel die Aufmerksamkeitsgewichte des Transformer-Encoders visualisieren, normalerweise mithilfe einer Heatmap. Ich verweise auf einige Blogs und habe das Gefühl, dass sie detailliert zusammengefasst wurden, wie zum Beispiel diesen: Python zeichnet Heatmap (Heatmap) Blog- CSDN blog_python Heatmap

Aber ich denke, ein Punkt ist nicht ganz klar. Nachdem ich es gelesen habe, weiß ich immer noch nicht, wie ich die Farbanpassung ändern soll. Bitte lesen Sie das offizielle Dokument: seaborn.heatmap – Dokumentation zu Seaborn 0.12.2

Um nur zu sagen, dass cmap zum Definieren des Farbraums verwendet werden kann:

cmap matplotlib Farbkartenname oder -objekt oder Liste von Farben, optional

Die Zuordnung von Datenwerten zum Farbraum. Wenn nicht angegeben, hängt die Standardeinstellung davon ab, ob sie centerfestgelegt ist.

Aber ich weiß überhaupt nicht, wie ich es verwenden soll. Konkret können Sie den Parameter cmap hinzufügen, zum Beispiel:

plot=sns.heatmap(p_array, cmap='Greens_r')

Den spezifischen Farbraum finden Sie hier: Choosing Colormaps in Matplotlib – Matplotlib 3.6.3-Dokumentation

Wenn wir absichtlich den falschen Parameter eingeben, werden in der Fehlermeldung alle möglichen Parameter abgefragt:

Unterstützte Werte sind „Accent“, „Accent_r“, „Blues“, „Blues_r“, „BrBG“, „BrBG_r“, „BuGn“, „BuGn_r“, „BuPu“, „BuPu_r“, „CMRmap“, „CMRmap_r“. , 'Dark2', 'Dark2_r', 'GnBu', 'GnBu_r', 'Greens', 'Greens_r', 'Greys', 'Greys_r', 'OrRd', 'OrRd_r', 'Oranges', 'Oranges_r', ' PRGn“, „PRGn_r“, „Paired“, „Paired_r“, „Pastel1“, „Pastel1_r“, „Pastel2“, „Pastel2_r“, „PiYG“, „PiYG_r“, „PuBu“, „PuBuGn“, „PuBuGn_r“ , 'PuBu_r', 'PuOr', 'PuOr_r', 'PuRd', 'PuRd_r', 'Purples', 'Purples_r', 'RdBu', 'RdBu_r',„RdGy“, „RdGy_r“, „RdPu“, „RdPu_r“, „RdYlBu“, „RdYlBu_r“, „RdYlGn“, „RdYlGn_r“, „Reds“, „Reds_r“, „Set1“, „Set1_r“, „Set2“. ', 'Set2_r', 'Set3', 'Set3_r', 'Spectral', 'Spectral_r', 'Wistia', 'Wistia_r', 'YlGn', 'YlGnBu', 'YlGnBu_r', 'YlGn_r', 'YlOrBr', 'YlOrBr_r', 'YlOrRd', 'YlOrRd_r', 'afmhot', 'afmhot_r', 'autumn', 'autumn_r', 'binary', 'binary_r', 'bone', 'bone_r', 'brg', 'brg_r ', 'bwr', 'bwr_r', 'cividis', 'cividis_r', 'cool', 'cool_r', 'coolwarm', 'coolwarm_r', 'copper','copper_r', 'crest', 'crest_r', 'cubehelix', 'cubehelix_r', 'flag', 'flag_r', 'flare', 'flare_r', 'gist_earth', 'gist_earth_r', 'gist_gray', 'gist_gray_r ', 'gist_heat', 'gist_heat_r', 'gist_ncar', 'gist_ncar_r', 'gist_rainbow', 'gist_rainbow_r', 'gist_stern', 'gist_stern_r', 'gist_yarg', 'gist_yarg_r', 'gnuplot', 'gnuplot2', 'gnuplot2_r', 'gnuplot_r', 'gray', 'gray_r', 'hot', 'hot_r', 'hsv', 'hsv_r', 'icefire', 'icefire_r', 'inferno', 'inferno_r', 'jet ', 'jet_r', 'magma', 'magma_r', 'mako', 'mako_r','nipy_spectral', 'nipy_spectral_r', 'ocean', 'ocean_r', 'pink', 'pink_r', 'plasma', 'plasma_r', 'prism', 'prism_r', 'rainbow', 'rainbow_r', 'rocket ', 'rocket_r', 'seismic', 'seismic_r', 'spring', 'spring_r', 'summer', 'summer_r', 'tab10', 'tab10_r', 'tab20', 'tab20_r', 'tab20b', 'tab20b_r', 'tab20c', 'tab20c_r', 'terrain', 'terrain_r', 'turbo', 'turbo_r', 'twilight', 'twilight_r', 'twilight_shifted', 'twilight_shifted_r', 'viridis', 'viridis_r ', 'vlag', 'vlag_r', 'winter', 'winter_r''nipy_spectral_r', 'ocean', 'ocean_r', 'pink', 'pink_r', 'plasma', 'plasma_r', 'prism', 'prism_r', 'rainbow', 'rainbow_r', 'rocket', 'rocket_r' ', 'seismisch', 'seismische_r', 'spring', 'spring_r', 'summer', 'summer_r', 'tab10', 'tab10_r', 'tab20', 'tab20_r', 'tab20b', 'tab20b_r', 'tab20c', 'tab20c_r', 'terrain', 'terrain_r', 'turbo', 'turbo_r', 'twilight', 'twilight_r', 'twilight_shifted', 'twilight_shifted_r', 'viridis', 'viridis_r', 'vlag ', 'vlag_r', 'winter', 'winter_r''nipy_spectral_r', 'ocean', 'ocean_r', 'pink', 'pink_r', 'plasma', 'plasma_r', 'prism', 'prism_r', 'rainbow', 'rainbow_r', 'rocket', 'rocket_r' ', 'seismisch', 'seismische_r', 'spring', 'spring_r', 'summer', 'summer_r', 'tab10', 'tab10_r', 'tab20', 'tab20_r', 'tab20b', 'tab20b_r', 'tab20c', 'tab20c_r', 'terrain', 'terrain_r', 'turbo', 'turbo_r', 'twilight', 'twilight_r', 'twilight_shifted', 'twilight_shifted_r', 'viridis', 'viridis_r', 'vlag ', 'vlag_r', 'winter', 'winter_r'„pink_r“, „plasma“, „plasma_r“, „prism“, „prism_r“, „rainbow“, „rainbow_r“, „rocket“, „rocket_r“, „seismic“, „seismic_r“, „spring“, „spring_r“. ', 'summer', 'summer_r', 'tab10', 'tab10_r', 'tab20', 'tab20_r', 'tab20b', 'tab20b_r', 'tab20c', 'tab20c_r', 'terrain', 'terrain_r', 'turbo', 'turbo_r', 'twilight', 'twilight_r', 'twilight_shifted', 'twilight_shifted_r', 'viridis', 'viridis_r', 'vlag', 'vlag_r', 'winter', 'winter_r'„pink_r“, „plasma“, „plasma_r“, „prism“, „prism_r“, „rainbow“, „rainbow_r“, „rocket“, „rocket_r“, „seismic“, „seismic_r“, „spring“, „spring_r“. ', 'summer', 'summer_r', 'tab10', 'tab10_r', 'tab20', 'tab20_r', 'tab20b', 'tab20b_r', 'tab20c', 'tab20c_r', 'terrain', 'terrain_r', 'turbo', 'turbo_r', 'twilight', 'twilight_r', 'twilight_shifted', 'twilight_shifted_r', 'viridis', 'viridis_r', 'vlag', 'vlag_r', 'winter', 'winter_r''rocket', 'rocket_r', 'seismic', 'seismic_r', 'spring', 'spring_r', 'summer', 'summer_r', 'tab10', 'tab10_r', 'tab20', 'tab20_r', 'tab20b ', 'tab20b_r', 'tab20c', 'tab20c_r', 'terrain', 'terrain_r', 'turbo', 'turbo_r', 'twilight', 'twilight_r', 'twilight_shifted', 'twilight_shifted_r', 'viridis', 'viridis_r', 'vlag', 'vlag_r', 'winter', 'winter_r''rocket', 'rocket_r', 'seismic', 'seismic_r', 'spring', 'spring_r', 'summer', 'summer_r', 'tab10', 'tab10_r', 'tab20', 'tab20_r', 'tab20b ', 'tab20b_r', 'tab20c', 'tab20c_r', 'terrain', 'terrain_r', 'turbo', 'turbo_r', 'twilight', 'twilight_r', 'twilight_shifted', 'twilight_shifted_r', 'viridis', 'viridis_r', 'vlag', 'vlag_r', 'winter', 'winter_r''tab20c', 'tab20c_r', 'terrain', 'terrain_r', 'turbo', 'turbo_r', 'twilight', 'twilight_r', 'twilight_shifted', 'twilight_shifted_r', 'viridis', 'viridis_r', 'vlag ', 'vlag_r', 'winter', 'winter_r''tab20c', 'tab20c_r', 'terrain', 'terrain_r', 'turbo', 'turbo_r', 'twilight', 'twilight_r', 'twilight_shifted', 'twilight_shifted_r', 'viridis', 'viridis_r', 'vlag ', 'vlag_r', 'winter', 'winter_r'

Zum Schluss fügen Sie noch ein Bild bei, das ich gemacht habe:

 

おすすめ

転載: blog.csdn.net/qysh123/article/details/128871826