{{ message }}
-
Notifications
You must be signed in to change notification settings - Fork 62
Expand file tree
/
Copy pathcm.html
More file actions
528 lines (428 loc) · 50.3 KB
/
Copy pathcm.html
File metadata and controls
528 lines (428 loc) · 50.3 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"
"http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
<html xmlns="http://www.w3.org/1999/xhtml">
<head>
<meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
<title>matplotlib.cm — Matplotlib 2.2.2 documentation</title>
<link rel="stylesheet" href="../../_static/mpl.css"
type="text/css" />
<link rel="stylesheet" href="../../_static/pygments.css"
type="text/css" />
<link rel="stylesheet" href="../../_static/gallery.css" type="text/css" />
<script type="text/javascript">
var DOCUMENTATION_OPTIONS = {
URL_ROOT: '../../',
VERSION: '2.2.2',
COLLAPSE_INDEX: false,
FILE_SUFFIX: '.html',
HAS_SOURCE: true,
SOURCELINK_SUFFIX: '.txt'
};
</script>
<script type="text/javascript" src="../../_static/jquery.js"></script>
<script type="text/javascript" src="../../_static/underscore.js"></script>
<script type="text/javascript" src="../../_static/doctools.js"></script>
<link rel="search" type="application/opensearchdescription+xml"
title="Search within Matplotlib 2.2.2 documentation"
href="../../_static/opensearch.xml"/>
<link rel="shortcut icon" href="../../_static/favicon.ico"/>
<link rel="index" title="Index" href="../../genindex.html"
/>
<link rel="search" title="Search" href="../../search.html"
/>
<link rel="top" title="Matplotlib 2.2.2 documentation" href="../../index.html" />
<link rel="up" title="matplotlib" href="../matplotlib.html" />
<link rel="canonical" href="https://matplotlib.org/3.4.3/_modules/matplotlib/cm.html" />
<script data-domain="matplotlib.org" defer="defer" src="https://views.scientific-python.org/js/script.js"></script>
</head>
<body>
<div id="unreleased-message"> You are reading an old version of the documentation (v2.2.2). For the latest version see <a href="/stable/">https://matplotlib.org/stable/</a> </div>
<div style="background-color: white; text-align: left; padding: 10px 10px 15px 15px; position: relative;">
<a href="../../index.html">
<div style="float: left; position: absolute; width: 496px; bottom: 0; padding-bottom: 24px"><span style="float: right; color: #789; background: white">Version 2.2.2</span></div>
<img src="../../_static/logo2.png" height="125px" border="0" alt="matplotlib"/></a>
<!-- The "Fork me on github" ribbon -->
<div id="forkongithub"><a href="https://github.com/matplotlib/matplotlib">Fork me on GitHub</a></div>
</div>
<div class="related">
<h3>Navigation</h3>
<ul>
<li class="right" style="margin-right: 10px">
<a href="../../genindex.html" title="General Index"
accesskey="I">index</a></li>
<li class="right" >
<a href="../../py-modindex.html" title="Python Module Index"
>modules</a> |</li>
<li><a href="../../index.html">home</a>| </li>
<li><a href="../../gallery/index.html">examples</a>| </li>
<li><a href="../../tutorials/index.html">tutorials</a>| </li>
<li><a href="../../api/pyplot_summary.html">pyplot</a>| </li>
<li><a href="../../contents.html">docs</a> »</li>
<li><a href="../index.html" >Module code</a> »</li>
<li><a href="../matplotlib.html" accesskey="U">matplotlib</a> »</li>
</ul>
</div>
<div class="sphinxsidebar">
<div class="sphinxsidebarwrapper">
<div id="searchbox" style="display: none" role="search">
<h3>Quick search</h3>
<div class="searchformwrapper">
<form class="search" action="../../search.html" method="get">
<input type="text" name="q" />
<input type="submit" value="Go" />
<input type="hidden" name="check_keywords" value="yes" />
<input type="hidden" name="area" value="default" />
</form>
</div>
</div>
<script type="text/javascript">$('#searchbox').show(0);</script><div class="relations">
<h3>Related Topics</h3>
<ul>
<li><a href="../../contents.html">Documentation overview</a><ul>
<li><a href="../index.html">Module code</a><ul>
<li><a href="../matplotlib.html">matplotlib</a><ul>
</ul></li>
</ul></li>
</ul></li>
</ul>
</div>
</div>
</div>
<div class="document">
<div class="documentwrapper">
<div class="bodywrapper">
<div class="body">
<h1>Source code for matplotlib.cm</h1><div class="highlight"><pre>
<span></span><span class="sd">"""</span>
<span class="sd">Builtin colormaps, colormap handling utilities, and the `ScalarMappable` mixin.</span>
<span class="sd">See :doc:`/gallery/color/colormap_reference` for a list of builtin colormaps.</span>
<span class="sd">See :doc:`/tutorials/colors/colormaps` for an in-depth discussion of colormaps.</span>
<span class="sd">"""</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="p">(</span><span class="n">absolute_import</span><span class="p">,</span> <span class="n">division</span><span class="p">,</span> <span class="n">print_function</span><span class="p">,</span>
<span class="n">unicode_literals</span><span class="p">)</span>
<span class="kn">import</span> <span class="nn">six</span>
<span class="kn">import</span> <span class="nn">numpy</span> <span class="k">as</span> <span class="nn">np</span>
<span class="kn">from</span> <span class="nn">numpy</span> <span class="k">import</span> <span class="n">ma</span>
<span class="kn">import</span> <span class="nn">matplotlib</span> <span class="k">as</span> <span class="nn">mpl</span>
<span class="kn">import</span> <span class="nn">matplotlib.colors</span> <span class="k">as</span> <span class="nn">colors</span>
<span class="kn">import</span> <span class="nn">matplotlib.cbook</span> <span class="k">as</span> <span class="nn">cbook</span>
<span class="kn">from</span> <span class="nn">matplotlib._cm</span> <span class="k">import</span> <span class="n">datad</span>
<span class="kn">from</span> <span class="nn">matplotlib._cm_listed</span> <span class="k">import</span> <span class="n">cmaps</span> <span class="k">as</span> <span class="n">cmaps_listed</span>
<span class="n">cmap_d</span> <span class="o">=</span> <span class="p">{}</span>
<span class="c1"># reverse all the colormaps.</span>
<span class="c1"># reversed colormaps have '_r' appended to the name.</span>
<span class="k">def</span> <span class="nf">_reverser</span><span class="p">(</span><span class="n">f</span><span class="p">):</span>
<span class="k">def</span> <span class="nf">freversed</span><span class="p">(</span><span class="n">x</span><span class="p">):</span>
<span class="k">return</span> <span class="n">f</span><span class="p">(</span><span class="mi">1</span> <span class="o">-</span> <span class="n">x</span><span class="p">)</span>
<span class="k">return</span> <span class="n">freversed</span>
<div class="viewcode-block" id="revcmap"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.revcmap">[docs]</a><span class="k">def</span> <span class="nf">revcmap</span><span class="p">(</span><span class="n">data</span><span class="p">):</span>
<span class="sd">"""Can only handle specification *data* in dictionary format."""</span>
<span class="n">data_r</span> <span class="o">=</span> <span class="p">{}</span>
<span class="k">for</span> <span class="n">key</span><span class="p">,</span> <span class="n">val</span> <span class="ow">in</span> <span class="n">six</span><span class="o">.</span><span class="n">iteritems</span><span class="p">(</span><span class="n">data</span><span class="p">):</span>
<span class="k">if</span> <span class="n">callable</span><span class="p">(</span><span class="n">val</span><span class="p">):</span>
<span class="n">valnew</span> <span class="o">=</span> <span class="n">_reverser</span><span class="p">(</span><span class="n">val</span><span class="p">)</span>
<span class="c1"># This doesn't work: lambda x: val(1-x)</span>
<span class="c1"># The same "val" (the first one) is used</span>
<span class="c1"># each time, so the colors are identical</span>
<span class="c1"># and the result is shades of gray.</span>
<span class="k">else</span><span class="p">:</span>
<span class="c1"># Flip x and exchange the y values facing x = 0 and x = 1.</span>
<span class="n">valnew</span> <span class="o">=</span> <span class="p">[(</span><span class="mf">1.0</span> <span class="o">-</span> <span class="n">x</span><span class="p">,</span> <span class="n">y1</span><span class="p">,</span> <span class="n">y0</span><span class="p">)</span> <span class="k">for</span> <span class="n">x</span><span class="p">,</span> <span class="n">y0</span><span class="p">,</span> <span class="n">y1</span> <span class="ow">in</span> <span class="nb">reversed</span><span class="p">(</span><span class="n">val</span><span class="p">)]</span>
<span class="n">data_r</span><span class="p">[</span><span class="n">key</span><span class="p">]</span> <span class="o">=</span> <span class="n">valnew</span>
<span class="k">return</span> <span class="n">data_r</span></div>
<span class="k">def</span> <span class="nf">_reverse_cmap_spec</span><span class="p">(</span><span class="n">spec</span><span class="p">):</span>
<span class="sd">"""Reverses cmap specification *spec*, can handle both dict and tuple</span>
<span class="sd"> type specs."""</span>
<span class="k">if</span> <span class="s1">'listed'</span> <span class="ow">in</span> <span class="n">spec</span><span class="p">:</span>
<span class="k">return</span> <span class="p">{</span><span class="s1">'listed'</span><span class="p">:</span> <span class="n">spec</span><span class="p">[</span><span class="s1">'listed'</span><span class="p">][::</span><span class="o">-</span><span class="mi">1</span><span class="p">]}</span>
<span class="k">if</span> <span class="s1">'red'</span> <span class="ow">in</span> <span class="n">spec</span><span class="p">:</span>
<span class="k">return</span> <span class="n">revcmap</span><span class="p">(</span><span class="n">spec</span><span class="p">)</span>
<span class="k">else</span><span class="p">:</span>
<span class="n">revspec</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="nb">reversed</span><span class="p">(</span><span class="n">spec</span><span class="p">))</span>
<span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">revspec</span><span class="p">[</span><span class="mi">0</span><span class="p">])</span> <span class="o">==</span> <span class="mi">2</span><span class="p">:</span> <span class="c1"># e.g., (1, (1.0, 0.0, 1.0))</span>
<span class="n">revspec</span> <span class="o">=</span> <span class="p">[(</span><span class="mf">1.0</span> <span class="o">-</span> <span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">)</span> <span class="k">for</span> <span class="n">a</span><span class="p">,</span> <span class="n">b</span> <span class="ow">in</span> <span class="n">revspec</span><span class="p">]</span>
<span class="k">return</span> <span class="n">revspec</span>
<span class="k">def</span> <span class="nf">_generate_cmap</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">lutsize</span><span class="p">):</span>
<span class="sd">"""Generates the requested cmap from its *name*. The lut size is</span>
<span class="sd"> *lutsize*."""</span>
<span class="n">spec</span> <span class="o">=</span> <span class="n">datad</span><span class="p">[</span><span class="n">name</span><span class="p">]</span>
<span class="c1"># Generate the colormap object.</span>
<span class="k">if</span> <span class="s1">'red'</span> <span class="ow">in</span> <span class="n">spec</span><span class="p">:</span>
<span class="k">return</span> <span class="n">colors</span><span class="o">.</span><span class="n">LinearSegmentedColormap</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">spec</span><span class="p">,</span> <span class="n">lutsize</span><span class="p">)</span>
<span class="k">elif</span> <span class="s1">'listed'</span> <span class="ow">in</span> <span class="n">spec</span><span class="p">:</span>
<span class="k">return</span> <span class="n">colors</span><span class="o">.</span><span class="n">ListedColormap</span><span class="p">(</span><span class="n">spec</span><span class="p">[</span><span class="s1">'listed'</span><span class="p">],</span> <span class="n">name</span><span class="p">)</span>
<span class="k">else</span><span class="p">:</span>
<span class="k">return</span> <span class="n">colors</span><span class="o">.</span><span class="n">LinearSegmentedColormap</span><span class="o">.</span><span class="n">from_list</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">spec</span><span class="p">,</span> <span class="n">lutsize</span><span class="p">)</span>
<span class="n">LUTSIZE</span> <span class="o">=</span> <span class="n">mpl</span><span class="o">.</span><span class="n">rcParams</span><span class="p">[</span><span class="s1">'image.lut'</span><span class="p">]</span>
<span class="c1"># Generate the reversed specifications (all at once, to avoid</span>
<span class="c1"># modify-when-iterating).</span>
<span class="n">datad</span><span class="o">.</span><span class="n">update</span><span class="p">({</span><span class="n">cmapname</span> <span class="o">+</span> <span class="s1">'_r'</span><span class="p">:</span> <span class="n">_reverse_cmap_spec</span><span class="p">(</span><span class="n">spec</span><span class="p">)</span>
<span class="k">for</span> <span class="n">cmapname</span><span class="p">,</span> <span class="n">spec</span> <span class="ow">in</span> <span class="n">six</span><span class="o">.</span><span class="n">iteritems</span><span class="p">(</span><span class="n">datad</span><span class="p">)})</span>
<span class="c1"># Precache the cmaps with ``lutsize = LUTSIZE``.</span>
<span class="c1"># Also add the reversed ones added in the section above:</span>
<span class="k">for</span> <span class="n">cmapname</span> <span class="ow">in</span> <span class="n">datad</span><span class="p">:</span>
<span class="n">cmap_d</span><span class="p">[</span><span class="n">cmapname</span><span class="p">]</span> <span class="o">=</span> <span class="n">_generate_cmap</span><span class="p">(</span><span class="n">cmapname</span><span class="p">,</span> <span class="n">LUTSIZE</span><span class="p">)</span>
<span class="n">cmap_d</span><span class="o">.</span><span class="n">update</span><span class="p">(</span><span class="n">cmaps_listed</span><span class="p">)</span>
<span class="nb">locals</span><span class="p">()</span><span class="o">.</span><span class="n">update</span><span class="p">(</span><span class="n">cmap_d</span><span class="p">)</span>
<span class="c1"># Continue with definitions ...</span>
<div class="viewcode-block" id="register_cmap"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.register_cmap">[docs]</a><span class="k">def</span> <span class="nf">register_cmap</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span> <span class="n">cmap</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span> <span class="n">data</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span> <span class="n">lut</span><span class="o">=</span><span class="kc">None</span><span class="p">):</span>
<span class="sd">"""</span>
<span class="sd"> Add a colormap to the set recognized by :func:`get_cmap`.</span>
<span class="sd"> It can be used in two ways::</span>
<span class="sd"> register_cmap(name='swirly', cmap=swirly_cmap)</span>
<span class="sd"> register_cmap(name='choppy', data=choppydata, lut=128)</span>
<span class="sd"> In the first case, *cmap* must be a :class:`matplotlib.colors.Colormap`</span>
<span class="sd"> instance. The *name* is optional; if absent, the name will</span>
<span class="sd"> be the :attr:`~matplotlib.colors.Colormap.name` attribute of the *cmap*.</span>
<span class="sd"> In the second case, the three arguments are passed to</span>
<span class="sd"> the :class:`~matplotlib.colors.LinearSegmentedColormap` initializer,</span>
<span class="sd"> and the resulting colormap is registered.</span>
<span class="sd"> """</span>
<span class="k">if</span> <span class="n">name</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">try</span><span class="p">:</span>
<span class="n">name</span> <span class="o">=</span> <span class="n">cmap</span><span class="o">.</span><span class="n">name</span>
<span class="k">except</span> <span class="ne">AttributeError</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">"Arguments must include a name or a Colormap"</span><span class="p">)</span>
<span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">six</span><span class="o">.</span><span class="n">string_types</span><span class="p">):</span>
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">"Colormap name must be a string"</span><span class="p">)</span>
<span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">cmap</span><span class="p">,</span> <span class="n">colors</span><span class="o">.</span><span class="n">Colormap</span><span class="p">):</span>
<span class="n">cmap_d</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="n">cmap</span>
<span class="k">return</span>
<span class="c1"># For the remainder, let exceptions propagate.</span>
<span class="k">if</span> <span class="n">lut</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="n">lut</span> <span class="o">=</span> <span class="n">mpl</span><span class="o">.</span><span class="n">rcParams</span><span class="p">[</span><span class="s1">'image.lut'</span><span class="p">]</span>
<span class="n">cmap</span> <span class="o">=</span> <span class="n">colors</span><span class="o">.</span><span class="n">LinearSegmentedColormap</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">data</span><span class="p">,</span> <span class="n">lut</span><span class="p">)</span>
<span class="n">cmap_d</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="n">cmap</span></div>
<div class="viewcode-block" id="get_cmap"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.get_cmap">[docs]</a><span class="k">def</span> <span class="nf">get_cmap</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span> <span class="n">lut</span><span class="o">=</span><span class="kc">None</span><span class="p">):</span>
<span class="sd">"""</span>
<span class="sd"> Get a colormap instance, defaulting to rc values if *name* is None.</span>
<span class="sd"> Colormaps added with :func:`register_cmap` take precedence over</span>
<span class="sd"> built-in colormaps.</span>
<span class="sd"> If *name* is a :class:`matplotlib.colors.Colormap` instance, it will be</span>
<span class="sd"> returned.</span>
<span class="sd"> If *lut* is not None it must be an integer giving the number of</span>
<span class="sd"> entries desired in the lookup table, and *name* must be a standard</span>
<span class="sd"> mpl colormap name.</span>
<span class="sd"> """</span>
<span class="k">if</span> <span class="n">name</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="n">name</span> <span class="o">=</span> <span class="n">mpl</span><span class="o">.</span><span class="n">rcParams</span><span class="p">[</span><span class="s1">'image.cmap'</span><span class="p">]</span>
<span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">colors</span><span class="o">.</span><span class="n">Colormap</span><span class="p">):</span>
<span class="k">return</span> <span class="n">name</span>
<span class="k">if</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">cmap_d</span><span class="p">:</span>
<span class="k">if</span> <span class="n">lut</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">return</span> <span class="n">cmap_d</span><span class="p">[</span><span class="n">name</span><span class="p">]</span>
<span class="k">else</span><span class="p">:</span>
<span class="k">return</span> <span class="n">cmap_d</span><span class="p">[</span><span class="n">name</span><span class="p">]</span><span class="o">.</span><span class="n">_resample</span><span class="p">(</span><span class="n">lut</span><span class="p">)</span>
<span class="k">else</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span>
<span class="s2">"Colormap </span><span class="si">%s</span><span class="s2"> is not recognized. Possible values are: </span><span class="si">%s</span><span class="s2">"</span>
<span class="o">%</span> <span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="s1">', '</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="nb">sorted</span><span class="p">(</span><span class="n">cmap_d</span><span class="p">))))</span></div>
<div class="viewcode-block" id="ScalarMappable"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.ScalarMappable">[docs]</a><span class="k">class</span> <span class="nc">ScalarMappable</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span>
<span class="sd">"""</span>
<span class="sd"> This is a mixin class to support scalar data to RGBA mapping.</span>
<span class="sd"> The ScalarMappable makes use of data normalization before returning</span>
<span class="sd"> RGBA colors from the given colormap.</span>
<span class="sd"> """</span>
<span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">norm</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span> <span class="n">cmap</span><span class="o">=</span><span class="kc">None</span><span class="p">):</span>
<span class="sa">r</span><span class="sd">"""</span>
<span class="sd"> Parameters</span>
<span class="sd"> ----------</span>
<span class="sd"> norm : :class:`matplotlib.colors.Normalize` instance</span>
<span class="sd"> The normalizing object which scales data, typically into the</span>
<span class="sd"> interval ``[0, 1]``.</span>
<span class="sd"> If *None*, *norm* defaults to a *colors.Normalize* object which</span>
<span class="sd"> initializes its scaling based on the first data processed.</span>
<span class="sd"> cmap : str or :class:`~matplotlib.colors.Colormap` instance</span>
<span class="sd"> The colormap used to map normalized data values to RGBA colors.</span>
<span class="sd"> """</span>
<span class="bp">self</span><span class="o">.</span><span class="n">callbacksSM</span> <span class="o">=</span> <span class="n">cbook</span><span class="o">.</span><span class="n">CallbackRegistry</span><span class="p">()</span>
<span class="k">if</span> <span class="n">cmap</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="n">cmap</span> <span class="o">=</span> <span class="n">get_cmap</span><span class="p">()</span>
<span class="k">if</span> <span class="n">norm</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="n">norm</span> <span class="o">=</span> <span class="n">colors</span><span class="o">.</span><span class="n">Normalize</span><span class="p">()</span>
<span class="bp">self</span><span class="o">.</span><span class="n">_A</span> <span class="o">=</span> <span class="kc">None</span>
<span class="c1">#: The Normalization instance of this ScalarMappable.</span>
<span class="bp">self</span><span class="o">.</span><span class="n">norm</span> <span class="o">=</span> <span class="n">norm</span>
<span class="c1">#: The Colormap instance of this ScalarMappable.</span>
<span class="bp">self</span><span class="o">.</span><span class="n">cmap</span> <span class="o">=</span> <span class="n">get_cmap</span><span class="p">(</span><span class="n">cmap</span><span class="p">)</span>
<span class="c1">#: The last colorbar associated with this ScalarMappable. May be None.</span>
<span class="bp">self</span><span class="o">.</span><span class="n">colorbar</span> <span class="o">=</span> <span class="kc">None</span>
<span class="bp">self</span><span class="o">.</span><span class="n">update_dict</span> <span class="o">=</span> <span class="p">{</span><span class="s1">'array'</span><span class="p">:</span> <span class="kc">False</span><span class="p">}</span>
<div class="viewcode-block" id="ScalarMappable.to_rgba"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.ScalarMappable.to_rgba">[docs]</a> <span class="k">def</span> <span class="nf">to_rgba</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">x</span><span class="p">,</span> <span class="n">alpha</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span> <span class="nb">bytes</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> <span class="n">norm</span><span class="o">=</span><span class="kc">True</span><span class="p">):</span>
<span class="sd">"""</span>
<span class="sd"> Return a normalized rgba array corresponding to *x*.</span>
<span class="sd"> In the normal case, *x* is a 1-D or 2-D sequence of scalars, and</span>
<span class="sd"> the corresponding ndarray of rgba values will be returned,</span>
<span class="sd"> based on the norm and colormap set for this ScalarMappable.</span>
<span class="sd"> There is one special case, for handling images that are already</span>
<span class="sd"> rgb or rgba, such as might have been read from an image file.</span>
<span class="sd"> If *x* is an ndarray with 3 dimensions,</span>
<span class="sd"> and the last dimension is either 3 or 4, then it will be</span>
<span class="sd"> treated as an rgb or rgba array, and no mapping will be done.</span>
<span class="sd"> The array can be uint8, or it can be floating point with</span>
<span class="sd"> values in the 0-1 range; otherwise a ValueError will be raised.</span>
<span class="sd"> If it is a masked array, the mask will be ignored.</span>
<span class="sd"> If the last dimension is 3, the *alpha* kwarg (defaulting to 1)</span>
<span class="sd"> will be used to fill in the transparency. If the last dimension</span>
<span class="sd"> is 4, the *alpha* kwarg is ignored; it does not</span>
<span class="sd"> replace the pre-existing alpha. A ValueError will be raised</span>
<span class="sd"> if the third dimension is other than 3 or 4.</span>
<span class="sd"> In either case, if *bytes* is *False* (default), the rgba</span>
<span class="sd"> array will be floats in the 0-1 range; if it is *True*,</span>
<span class="sd"> the returned rgba array will be uint8 in the 0 to 255 range.</span>
<span class="sd"> If norm is False, no normalization of the input data is</span>
<span class="sd"> performed, and it is assumed to be in the range (0-1).</span>
<span class="sd"> """</span>
<span class="c1"># First check for special case, image input:</span>
<span class="k">try</span><span class="p">:</span>
<span class="k">if</span> <span class="n">x</span><span class="o">.</span><span class="n">ndim</span> <span class="o">==</span> <span class="mi">3</span><span class="p">:</span>
<span class="k">if</span> <span class="n">x</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span> <span class="o">==</span> <span class="mi">3</span><span class="p">:</span>
<span class="k">if</span> <span class="n">alpha</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="n">alpha</span> <span class="o">=</span> <span class="mi">1</span>
<span class="k">if</span> <span class="n">x</span><span class="o">.</span><span class="n">dtype</span> <span class="o">==</span> <span class="n">np</span><span class="o">.</span><span class="n">uint8</span><span class="p">:</span>
<span class="n">alpha</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">uint8</span><span class="p">(</span><span class="n">alpha</span> <span class="o">*</span> <span class="mi">255</span><span class="p">)</span>
<span class="n">m</span><span class="p">,</span> <span class="n">n</span> <span class="o">=</span> <span class="n">x</span><span class="o">.</span><span class="n">shape</span><span class="p">[:</span><span class="mi">2</span><span class="p">]</span>
<span class="n">xx</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">empty</span><span class="p">(</span><span class="n">shape</span><span class="o">=</span><span class="p">(</span><span class="n">m</span><span class="p">,</span> <span class="n">n</span><span class="p">,</span> <span class="mi">4</span><span class="p">),</span> <span class="n">dtype</span><span class="o">=</span><span class="n">x</span><span class="o">.</span><span class="n">dtype</span><span class="p">)</span>
<span class="n">xx</span><span class="p">[:,</span> <span class="p">:,</span> <span class="p">:</span><span class="mi">3</span><span class="p">]</span> <span class="o">=</span> <span class="n">x</span>
<span class="n">xx</span><span class="p">[:,</span> <span class="p">:,</span> <span class="mi">3</span><span class="p">]</span> <span class="o">=</span> <span class="n">alpha</span>
<span class="k">elif</span> <span class="n">x</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span> <span class="o">==</span> <span class="mi">4</span><span class="p">:</span>
<span class="n">xx</span> <span class="o">=</span> <span class="n">x</span>
<span class="k">else</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">"third dimension must be 3 or 4"</span><span class="p">)</span>
<span class="k">if</span> <span class="n">xx</span><span class="o">.</span><span class="n">dtype</span><span class="o">.</span><span class="n">kind</span> <span class="o">==</span> <span class="s1">'f'</span><span class="p">:</span>
<span class="k">if</span> <span class="n">norm</span> <span class="ow">and</span> <span class="n">xx</span><span class="o">.</span><span class="n">max</span><span class="p">()</span> <span class="o">></span> <span class="mi">1</span> <span class="ow">or</span> <span class="n">xx</span><span class="o">.</span><span class="n">min</span><span class="p">()</span> <span class="o"><</span> <span class="mi">0</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">"Floating point image RGB values "</span>
<span class="s2">"must be in the 0..1 range."</span><span class="p">)</span>
<span class="k">if</span> <span class="nb">bytes</span><span class="p">:</span>
<span class="n">xx</span> <span class="o">=</span> <span class="p">(</span><span class="n">xx</span> <span class="o">*</span> <span class="mi">255</span><span class="p">)</span><span class="o">.</span><span class="n">astype</span><span class="p">(</span><span class="n">np</span><span class="o">.</span><span class="n">uint8</span><span class="p">)</span>
<span class="k">elif</span> <span class="n">xx</span><span class="o">.</span><span class="n">dtype</span> <span class="o">==</span> <span class="n">np</span><span class="o">.</span><span class="n">uint8</span><span class="p">:</span>
<span class="k">if</span> <span class="ow">not</span> <span class="nb">bytes</span><span class="p">:</span>
<span class="n">xx</span> <span class="o">=</span> <span class="n">xx</span><span class="o">.</span><span class="n">astype</span><span class="p">(</span><span class="n">np</span><span class="o">.</span><span class="n">float32</span><span class="p">)</span> <span class="o">/</span> <span class="mi">255</span>
<span class="k">else</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">"Image RGB array must be uint8 or "</span>
<span class="s2">"floating point; found </span><span class="si">%s</span><span class="s2">"</span> <span class="o">%</span> <span class="n">xx</span><span class="o">.</span><span class="n">dtype</span><span class="p">)</span>
<span class="k">return</span> <span class="n">xx</span>
<span class="k">except</span> <span class="ne">AttributeError</span><span class="p">:</span>
<span class="c1"># e.g., x is not an ndarray; so try mapping it</span>
<span class="k">pass</span>
<span class="c1"># This is the normal case, mapping a scalar array:</span>
<span class="n">x</span> <span class="o">=</span> <span class="n">ma</span><span class="o">.</span><span class="n">asarray</span><span class="p">(</span><span class="n">x</span><span class="p">)</span>
<span class="k">if</span> <span class="n">norm</span><span class="p">:</span>
<span class="n">x</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">norm</span><span class="p">(</span><span class="n">x</span><span class="p">)</span>
<span class="n">rgba</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">cmap</span><span class="p">(</span><span class="n">x</span><span class="p">,</span> <span class="n">alpha</span><span class="o">=</span><span class="n">alpha</span><span class="p">,</span> <span class="nb">bytes</span><span class="o">=</span><span class="nb">bytes</span><span class="p">)</span>
<span class="k">return</span> <span class="n">rgba</span></div>
<div class="viewcode-block" id="ScalarMappable.set_array"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.ScalarMappable.set_array">[docs]</a> <span class="k">def</span> <span class="nf">set_array</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">A</span><span class="p">):</span>
<span class="sd">"""Set the image array from numpy array *A*.</span>
<span class="sd"> .. ACCEPTS: ndarray</span>
<span class="sd"> Parameters</span>
<span class="sd"> ----------</span>
<span class="sd"> A : ndarray</span>
<span class="sd"> """</span>
<span class="bp">self</span><span class="o">.</span><span class="n">_A</span> <span class="o">=</span> <span class="n">A</span>
<span class="bp">self</span><span class="o">.</span><span class="n">update_dict</span><span class="p">[</span><span class="s1">'array'</span><span class="p">]</span> <span class="o">=</span> <span class="kc">True</span></div>
<div class="viewcode-block" id="ScalarMappable.get_array"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.ScalarMappable.get_array">[docs]</a> <span class="k">def</span> <span class="nf">get_array</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="s1">'Return the array'</span>
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_A</span></div>
<div class="viewcode-block" id="ScalarMappable.get_cmap"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.ScalarMappable.get_cmap">[docs]</a> <span class="k">def</span> <span class="nf">get_cmap</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="s1">'return the colormap'</span>
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">cmap</span></div>
<div class="viewcode-block" id="ScalarMappable.get_clim"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.ScalarMappable.get_clim">[docs]</a> <span class="k">def</span> <span class="nf">get_clim</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="s1">'return the min, max of the color limits for image scaling'</span>
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">norm</span><span class="o">.</span><span class="n">vmin</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">norm</span><span class="o">.</span><span class="n">vmax</span></div>
<div class="viewcode-block" id="ScalarMappable.set_clim"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.ScalarMappable.set_clim">[docs]</a> <span class="k">def</span> <span class="nf">set_clim</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">vmin</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span> <span class="n">vmax</span><span class="o">=</span><span class="kc">None</span><span class="p">):</span>
<span class="sd">"""</span>
<span class="sd"> set the norm limits for image scaling; if *vmin* is a length2</span>
<span class="sd"> sequence, interpret it as ``(vmin, vmax)`` which is used to</span>
<span class="sd"> support setp</span>
<span class="sd"> ACCEPTS: a length 2 sequence of floats; may be overridden in methods</span>
<span class="sd"> that have ``vmin`` and ``vmax`` kwargs.</span>
<span class="sd"> """</span>
<span class="k">if</span> <span class="n">vmax</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">try</span><span class="p">:</span>
<span class="n">vmin</span><span class="p">,</span> <span class="n">vmax</span> <span class="o">=</span> <span class="n">vmin</span>
<span class="k">except</span> <span class="p">(</span><span class="ne">TypeError</span><span class="p">,</span> <span class="ne">ValueError</span><span class="p">):</span>
<span class="k">pass</span>
<span class="k">if</span> <span class="n">vmin</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">norm</span><span class="o">.</span><span class="n">vmin</span> <span class="o">=</span> <span class="n">colors</span><span class="o">.</span><span class="n">_sanitize_extrema</span><span class="p">(</span><span class="n">vmin</span><span class="p">)</span>
<span class="k">if</span> <span class="n">vmax</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">norm</span><span class="o">.</span><span class="n">vmax</span> <span class="o">=</span> <span class="n">colors</span><span class="o">.</span><span class="n">_sanitize_extrema</span><span class="p">(</span><span class="n">vmax</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">changed</span><span class="p">()</span></div>
<div class="viewcode-block" id="ScalarMappable.set_cmap"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.ScalarMappable.set_cmap">[docs]</a> <span class="k">def</span> <span class="nf">set_cmap</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">cmap</span><span class="p">):</span>
<span class="sd">"""</span>
<span class="sd"> set the colormap for luminance data</span>
<span class="sd"> ACCEPTS: a colormap or registered colormap name</span>
<span class="sd"> """</span>
<span class="n">cmap</span> <span class="o">=</span> <span class="n">get_cmap</span><span class="p">(</span><span class="n">cmap</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">cmap</span> <span class="o">=</span> <span class="n">cmap</span>
<span class="bp">self</span><span class="o">.</span><span class="n">changed</span><span class="p">()</span></div>
<div class="viewcode-block" id="ScalarMappable.set_norm"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.ScalarMappable.set_norm">[docs]</a> <span class="k">def</span> <span class="nf">set_norm</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">norm</span><span class="p">):</span>
<span class="sd">"""Set the normalization instance.</span>
<span class="sd"> .. ACCEPTS: `.Normalize`</span>
<span class="sd"> Parameters</span>
<span class="sd"> ----------</span>
<span class="sd"> norm : `.Normalize`</span>
<span class="sd"> """</span>
<span class="k">if</span> <span class="n">norm</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="n">norm</span> <span class="o">=</span> <span class="n">colors</span><span class="o">.</span><span class="n">Normalize</span><span class="p">()</span>
<span class="bp">self</span><span class="o">.</span><span class="n">norm</span> <span class="o">=</span> <span class="n">norm</span>
<span class="bp">self</span><span class="o">.</span><span class="n">changed</span><span class="p">()</span></div>
<div class="viewcode-block" id="ScalarMappable.autoscale"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.ScalarMappable.autoscale">[docs]</a> <span class="k">def</span> <span class="nf">autoscale</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="sd">"""</span>
<span class="sd"> Autoscale the scalar limits on the norm instance using the</span>
<span class="sd"> current array</span>
<span class="sd"> """</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">_A</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">TypeError</span><span class="p">(</span><span class="s1">'You must first set_array for mappable'</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">norm</span><span class="o">.</span><span class="n">autoscale</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_A</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">changed</span><span class="p">()</span></div>
<div class="viewcode-block" id="ScalarMappable.autoscale_None"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.ScalarMappable.autoscale_None">[docs]</a> <span class="k">def</span> <span class="nf">autoscale_None</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="sd">"""</span>
<span class="sd"> Autoscale the scalar limits on the norm instance using the</span>
<span class="sd"> current array, changing only limits that are None</span>
<span class="sd"> """</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">_A</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">TypeError</span><span class="p">(</span><span class="s1">'You must first set_array for mappable'</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">norm</span><span class="o">.</span><span class="n">autoscale_None</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_A</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">changed</span><span class="p">()</span></div>
<div class="viewcode-block" id="ScalarMappable.add_checker"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.ScalarMappable.add_checker">[docs]</a> <span class="k">def</span> <span class="nf">add_checker</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">checker</span><span class="p">):</span>
<span class="sd">"""</span>
<span class="sd"> Add an entry to a dictionary of boolean flags</span>
<span class="sd"> that are set to True when the mappable is changed.</span>
<span class="sd"> """</span>
<span class="bp">self</span><span class="o">.</span><span class="n">update_dict</span><span class="p">[</span><span class="n">checker</span><span class="p">]</span> <span class="o">=</span> <span class="kc">False</span></div>
<div class="viewcode-block" id="ScalarMappable.check_update"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.ScalarMappable.check_update">[docs]</a> <span class="k">def</span> <span class="nf">check_update</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">checker</span><span class="p">):</span>
<span class="sd">"""</span>
<span class="sd"> If mappable has changed since the last check,</span>
<span class="sd"> return True; else return False</span>
<span class="sd"> """</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">update_dict</span><span class="p">[</span><span class="n">checker</span><span class="p">]:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">update_dict</span><span class="p">[</span><span class="n">checker</span><span class="p">]</span> <span class="o">=</span> <span class="kc">False</span>
<span class="k">return</span> <span class="kc">True</span>
<span class="k">return</span> <span class="kc">False</span></div>
<div class="viewcode-block" id="ScalarMappable.changed"><a class="viewcode-back" href="../../api/cm_api.html#matplotlib.cm.ScalarMappable.changed">[docs]</a> <span class="k">def</span> <span class="nf">changed</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="sd">"""</span>
<span class="sd"> Call this whenever the mappable is changed to notify all the</span>
<span class="sd"> callbackSM listeners to the 'changed' signal</span>
<span class="sd"> """</span>
<span class="bp">self</span><span class="o">.</span><span class="n">callbacksSM</span><span class="o">.</span><span class="n">process</span><span class="p">(</span><span class="s1">'changed'</span><span class="p">,</span> <span class="bp">self</span><span class="p">)</span>
<span class="k">for</span> <span class="n">key</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">update_dict</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">update_dict</span><span class="p">[</span><span class="n">key</span><span class="p">]</span> <span class="o">=</span> <span class="kc">True</span>
<span class="bp">self</span><span class="o">.</span><span class="n">stale</span> <span class="o">=</span> <span class="kc">True</span></div></div>
</pre></div>
</div>
</div>
</div>
<div class="clearer"></div>
</div>
<div class="footer">
© Copyright 2002 - 2012 John Hunter, Darren Dale, Eric Firing, Michael Droettboom and the Matplotlib development team; 2012 - 2018 The Matplotlib development team.
<br />
Last updated on Jun 28, 2018.
Created using
<ahref="http://sphinx-doc.org/">Sphinx</a> 1.7.5.
Doc version v2.2.2-101-g15e1eadd0.
</div>
</body>
<footer>
</footer>
</html>
You can’t perform that action at this time.