Last active
March 31, 2023 02:45
-
-
Save FedeMiorelli/640bbc66b2038a14802729e609abfe89 to your computer and use it in GitHub Desktop.
Turbo Colormap for Matplotlib
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # -*- coding: utf-8 -*- | |
| """ | |
| Created on 2019-08-22 09:37:36 | |
| @author: fmiorell | |
| """ | |
| # This script registers the "turbo" colormap to matplotlib, and the reversed version as "turbo_r" | |
| # Reference: https://ai.googleblog.com/2019/08/turbo-improved-rainbow-colormap-for.html | |
| import numpy as np | |
| import matplotlib.pyplot as plt | |
| turbo_colormap_data = np.array( | |
| [[0.18995,0.07176,0.23217], | |
| [0.19483,0.08339,0.26149], | |
| [0.19956,0.09498,0.29024], | |
| [0.20415,0.10652,0.31844], | |
| [0.20860,0.11802,0.34607], | |
| [0.21291,0.12947,0.37314], | |
| [0.21708,0.14087,0.39964], | |
| [0.22111,0.15223,0.42558], | |
| [0.22500,0.16354,0.45096], | |
| [0.22875,0.17481,0.47578], | |
| [0.23236,0.18603,0.50004], | |
| [0.23582,0.19720,0.52373], | |
| [0.23915,0.20833,0.54686], | |
| [0.24234,0.21941,0.56942], | |
| [0.24539,0.23044,0.59142], | |
| [0.24830,0.24143,0.61286], | |
| [0.25107,0.25237,0.63374], | |
| [0.25369,0.26327,0.65406], | |
| [0.25618,0.27412,0.67381], | |
| [0.25853,0.28492,0.69300], | |
| [0.26074,0.29568,0.71162], | |
| [0.26280,0.30639,0.72968], | |
| [0.26473,0.31706,0.74718], | |
| [0.26652,0.32768,0.76412], | |
| [0.26816,0.33825,0.78050], | |
| [0.26967,0.34878,0.79631], | |
| [0.27103,0.35926,0.81156], | |
| [0.27226,0.36970,0.82624], | |
| [0.27334,0.38008,0.84037], | |
| [0.27429,0.39043,0.85393], | |
| [0.27509,0.40072,0.86692], | |
| [0.27576,0.41097,0.87936], | |
| [0.27628,0.42118,0.89123], | |
| [0.27667,0.43134,0.90254], | |
| [0.27691,0.44145,0.91328], | |
| [0.27701,0.45152,0.92347], | |
| [0.27698,0.46153,0.93309], | |
| [0.27680,0.47151,0.94214], | |
| [0.27648,0.48144,0.95064], | |
| [0.27603,0.49132,0.95857], | |
| [0.27543,0.50115,0.96594], | |
| [0.27469,0.51094,0.97275], | |
| [0.27381,0.52069,0.97899], | |
| [0.27273,0.53040,0.98461], | |
| [0.27106,0.54015,0.98930], | |
| [0.26878,0.54995,0.99303], | |
| [0.26592,0.55979,0.99583], | |
| [0.26252,0.56967,0.99773], | |
| [0.25862,0.57958,0.99876], | |
| [0.25425,0.58950,0.99896], | |
| [0.24946,0.59943,0.99835], | |
| [0.24427,0.60937,0.99697], | |
| [0.23874,0.61931,0.99485], | |
| [0.23288,0.62923,0.99202], | |
| [0.22676,0.63913,0.98851], | |
| [0.22039,0.64901,0.98436], | |
| [0.21382,0.65886,0.97959], | |
| [0.20708,0.66866,0.97423], | |
| [0.20021,0.67842,0.96833], | |
| [0.19326,0.68812,0.96190], | |
| [0.18625,0.69775,0.95498], | |
| [0.17923,0.70732,0.94761], | |
| [0.17223,0.71680,0.93981], | |
| [0.16529,0.72620,0.93161], | |
| [0.15844,0.73551,0.92305], | |
| [0.15173,0.74472,0.91416], | |
| [0.14519,0.75381,0.90496], | |
| [0.13886,0.76279,0.89550], | |
| [0.13278,0.77165,0.88580], | |
| [0.12698,0.78037,0.87590], | |
| [0.12151,0.78896,0.86581], | |
| [0.11639,0.79740,0.85559], | |
| [0.11167,0.80569,0.84525], | |
| [0.10738,0.81381,0.83484], | |
| [0.10357,0.82177,0.82437], | |
| [0.10026,0.82955,0.81389], | |
| [0.09750,0.83714,0.80342], | |
| [0.09532,0.84455,0.79299], | |
| [0.09377,0.85175,0.78264], | |
| [0.09287,0.85875,0.77240], | |
| [0.09267,0.86554,0.76230], | |
| [0.09320,0.87211,0.75237], | |
| [0.09451,0.87844,0.74265], | |
| [0.09662,0.88454,0.73316], | |
| [0.09958,0.89040,0.72393], | |
| [0.10342,0.89600,0.71500], | |
| [0.10815,0.90142,0.70599], | |
| [0.11374,0.90673,0.69651], | |
| [0.12014,0.91193,0.68660], | |
| [0.12733,0.91701,0.67627], | |
| [0.13526,0.92197,0.66556], | |
| [0.14391,0.92680,0.65448], | |
| [0.15323,0.93151,0.64308], | |
| [0.16319,0.93609,0.63137], | |
| [0.17377,0.94053,0.61938], | |
| [0.18491,0.94484,0.60713], | |
| [0.19659,0.94901,0.59466], | |
| [0.20877,0.95304,0.58199], | |
| [0.22142,0.95692,0.56914], | |
| [0.23449,0.96065,0.55614], | |
| [0.24797,0.96423,0.54303], | |
| [0.26180,0.96765,0.52981], | |
| [0.27597,0.97092,0.51653], | |
| [0.29042,0.97403,0.50321], | |
| [0.30513,0.97697,0.48987], | |
| [0.32006,0.97974,0.47654], | |
| [0.33517,0.98234,0.46325], | |
| [0.35043,0.98477,0.45002], | |
| [0.36581,0.98702,0.43688], | |
| [0.38127,0.98909,0.42386], | |
| [0.39678,0.99098,0.41098], | |
| [0.41229,0.99268,0.39826], | |
| [0.42778,0.99419,0.38575], | |
| [0.44321,0.99551,0.37345], | |
| [0.45854,0.99663,0.36140], | |
| [0.47375,0.99755,0.34963], | |
| [0.48879,0.99828,0.33816], | |
| [0.50362,0.99879,0.32701], | |
| [0.51822,0.99910,0.31622], | |
| [0.53255,0.99919,0.30581], | |
| [0.54658,0.99907,0.29581], | |
| [0.56026,0.99873,0.28623], | |
| [0.57357,0.99817,0.27712], | |
| [0.58646,0.99739,0.26849], | |
| [0.59891,0.99638,0.26038], | |
| [0.61088,0.99514,0.25280], | |
| [0.62233,0.99366,0.24579], | |
| [0.63323,0.99195,0.23937], | |
| [0.64362,0.98999,0.23356], | |
| [0.65394,0.98775,0.22835], | |
| [0.66428,0.98524,0.22370], | |
| [0.67462,0.98246,0.21960], | |
| [0.68494,0.97941,0.21602], | |
| [0.69525,0.97610,0.21294], | |
| [0.70553,0.97255,0.21032], | |
| [0.71577,0.96875,0.20815], | |
| [0.72596,0.96470,0.20640], | |
| [0.73610,0.96043,0.20504], | |
| [0.74617,0.95593,0.20406], | |
| [0.75617,0.95121,0.20343], | |
| [0.76608,0.94627,0.20311], | |
| [0.77591,0.94113,0.20310], | |
| [0.78563,0.93579,0.20336], | |
| [0.79524,0.93025,0.20386], | |
| [0.80473,0.92452,0.20459], | |
| [0.81410,0.91861,0.20552], | |
| [0.82333,0.91253,0.20663], | |
| [0.83241,0.90627,0.20788], | |
| [0.84133,0.89986,0.20926], | |
| [0.85010,0.89328,0.21074], | |
| [0.85868,0.88655,0.21230], | |
| [0.86709,0.87968,0.21391], | |
| [0.87530,0.87267,0.21555], | |
| [0.88331,0.86553,0.21719], | |
| [0.89112,0.85826,0.21880], | |
| [0.89870,0.85087,0.22038], | |
| [0.90605,0.84337,0.22188], | |
| [0.91317,0.83576,0.22328], | |
| [0.92004,0.82806,0.22456], | |
| [0.92666,0.82025,0.22570], | |
| [0.93301,0.81236,0.22667], | |
| [0.93909,0.80439,0.22744], | |
| [0.94489,0.79634,0.22800], | |
| [0.95039,0.78823,0.22831], | |
| [0.95560,0.78005,0.22836], | |
| [0.96049,0.77181,0.22811], | |
| [0.96507,0.76352,0.22754], | |
| [0.96931,0.75519,0.22663], | |
| [0.97323,0.74682,0.22536], | |
| [0.97679,0.73842,0.22369], | |
| [0.98000,0.73000,0.22161], | |
| [0.98289,0.72140,0.21918], | |
| [0.98549,0.71250,0.21650], | |
| [0.98781,0.70330,0.21358], | |
| [0.98986,0.69382,0.21043], | |
| [0.99163,0.68408,0.20706], | |
| [0.99314,0.67408,0.20348], | |
| [0.99438,0.66386,0.19971], | |
| [0.99535,0.65341,0.19577], | |
| [0.99607,0.64277,0.19165], | |
| [0.99654,0.63193,0.18738], | |
| [0.99675,0.62093,0.18297], | |
| [0.99672,0.60977,0.17842], | |
| [0.99644,0.59846,0.17376], | |
| [0.99593,0.58703,0.16899], | |
| [0.99517,0.57549,0.16412], | |
| [0.99419,0.56386,0.15918], | |
| [0.99297,0.55214,0.15417], | |
| [0.99153,0.54036,0.14910], | |
| [0.98987,0.52854,0.14398], | |
| [0.98799,0.51667,0.13883], | |
| [0.98590,0.50479,0.13367], | |
| [0.98360,0.49291,0.12849], | |
| [0.98108,0.48104,0.12332], | |
| [0.97837,0.46920,0.11817], | |
| [0.97545,0.45740,0.11305], | |
| [0.97234,0.44565,0.10797], | |
| [0.96904,0.43399,0.10294], | |
| [0.96555,0.42241,0.09798], | |
| [0.96187,0.41093,0.09310], | |
| [0.95801,0.39958,0.08831], | |
| [0.95398,0.38836,0.08362], | |
| [0.94977,0.37729,0.07905], | |
| [0.94538,0.36638,0.07461], | |
| [0.94084,0.35566,0.07031], | |
| [0.93612,0.34513,0.06616], | |
| [0.93125,0.33482,0.06218], | |
| [0.92623,0.32473,0.05837], | |
| [0.92105,0.31489,0.05475], | |
| [0.91572,0.30530,0.05134], | |
| [0.91024,0.29599,0.04814], | |
| [0.90463,0.28696,0.04516], | |
| [0.89888,0.27824,0.04243], | |
| [0.89298,0.26981,0.03993], | |
| [0.88691,0.26152,0.03753], | |
| [0.88066,0.25334,0.03521], | |
| [0.87422,0.24526,0.03297], | |
| [0.86760,0.23730,0.03082], | |
| [0.86079,0.22945,0.02875], | |
| [0.85380,0.22170,0.02677], | |
| [0.84662,0.21407,0.02487], | |
| [0.83926,0.20654,0.02305], | |
| [0.83172,0.19912,0.02131], | |
| [0.82399,0.19182,0.01966], | |
| [0.81608,0.18462,0.01809], | |
| [0.80799,0.17753,0.01660], | |
| [0.79971,0.17055,0.01520], | |
| [0.79125,0.16368,0.01387], | |
| [0.78260,0.15693,0.01264], | |
| [0.77377,0.15028,0.01148], | |
| [0.76476,0.14374,0.01041], | |
| [0.75556,0.13731,0.00942], | |
| [0.74617,0.13098,0.00851], | |
| [0.73661,0.12477,0.00769], | |
| [0.72686,0.11867,0.00695], | |
| [0.71692,0.11268,0.00629], | |
| [0.70680,0.10680,0.00571], | |
| [0.69650,0.10102,0.00522], | |
| [0.68602,0.09536,0.00481], | |
| [0.67535,0.08980,0.00449], | |
| [0.66449,0.08436,0.00424], | |
| [0.65345,0.07902,0.00408], | |
| [0.64223,0.07380,0.00401], | |
| [0.63082,0.06868,0.00401], | |
| [0.61923,0.06367,0.00410], | |
| [0.60746,0.05878,0.00427], | |
| [0.59550,0.05399,0.00453], | |
| [0.58336,0.04931,0.00486], | |
| [0.57103,0.04474,0.00529], | |
| [0.55852,0.04028,0.00579], | |
| [0.54583,0.03593,0.00638], | |
| [0.53295,0.03169,0.00705], | |
| [0.51989,0.02756,0.00780], | |
| [0.50664,0.02354,0.00863], | |
| [0.49321,0.01963,0.00955], | |
| [0.47960,0.01583,0.01055]]) | |
| def RGBToPyCmap(rgbdata): | |
| nsteps = rgbdata.shape[0] | |
| stepaxis = np.linspace(0, 1, nsteps) | |
| rdata=[]; gdata=[]; bdata=[] | |
| for istep in range(nsteps): | |
| r = rgbdata[istep,0] | |
| g = rgbdata[istep,1] | |
| b = rgbdata[istep,2] | |
| rdata.append((stepaxis[istep], r, r)) | |
| gdata.append((stepaxis[istep], g, g)) | |
| bdata.append((stepaxis[istep], b, b)) | |
| mpl_data = {'red': rdata, | |
| 'green': gdata, | |
| 'blue': bdata} | |
| return mpl_data | |
| mpl_data = RGBToPyCmap(turbo_colormap_data) | |
| plt.register_cmap(name='turbo', data=mpl_data, lut=turbo_colormap_data.shape[0]) | |
| mpl_data_r = RGBToPyCmap(turbo_colormap_data[::-1,:]) | |
| plt.register_cmap(name='turbo_r', data=mpl_data_r, lut=turbo_colormap_data.shape[0]) | |
| if __name__=='__main__': | |
| XX, YY = np.meshgrid(np.linspace(0,1,100), np.linspace(0,1,100)) | |
| ZZ = np.sqrt(XX**2 + YY**2) | |
| plt.figure() | |
| plt.imshow(ZZ, cmap='turbo') | |
| plt.colorbar() | |
| plt.figure() | |
| plt.imshow(ZZ, cmap='turbo_r') | |
| plt.colorbar() | |
| plt.show() | |
it's missing an import numpy as np
@vnoel Fixed, thanks!
What's the license on this? :)
Thanks for this! For reference, here's more on the colormap by Google https://ai.googleblog.com/2019/08/turbo-improved-rainbow-colormap-for.html
Thank you @granttremblay, I added the reference at the beginning.
I have also added the registration of the reversed colormap "turbo_r" as per matplotlib tradition
Matplotlib 3.3.0 is now shipping Turbo colormap as part of the built-ins
I'm also wondering about the license. Would you let people use this file however they want to or is the allowed usage restricted?
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
it's missing an import numpy as np
thanks for this !