Support varying colors in a PatchCollection

Author: eric-wieser

test/test_patches.py

@@ -55,7 +55,7 @@ def plot():
         patches.append(polygon)
 
     colors = 100 * np.random.rand(len(patches))
-    p = PatchCollection(patches, cmap=mpl.cm.jet, alpha=0.4)
+    p = PatchCollection(patches, cmap=mpl.cm.viridis, alpha=0.4)
     p.set_array(np.array(colors))
     ax.add_collection(p)
 

test/test_patches_reference.tex

@@ -1,24 +1,25 @@
 \begin{tikzpicture}
 
-\definecolor{color0}{rgb}{0.12156862745098,0.466666666666667,0.705882352941177}
+\definecolor{color0}{rgb}{0.241237,0.296485,0.539709}
+\definecolor{color1}{rgb}{0.187231,0.414746,0.556547}
+\definecolor{color2}{rgb}{0.636902,0.856542,0.21662}
+\definecolor{color3}{rgb}{0.273006,0.20452,0.501721}
+\definecolor{color4}{rgb}{0.15627,0.489624,0.557936}
+\definecolor{color5}{rgb}{0.993248,0.906157,0.143936}
+\definecolor{color6}{rgb}{0.140536,0.530132,0.555659}
+\definecolor{color7}{rgb}{0.12138,0.629492,0.531973}
+\definecolor{color8}{rgb}{0.267004,0.004874,0.329415}
+\definecolor{color9}{rgb}{0.515992,0.831158,0.294279}
+\definecolor{color10}{rgb}{0.119699,0.61849,0.536347}
+\definecolor{color11}{rgb}{0.131172,0.555899,0.552459}
+\definecolor{color12}{rgb}{0.227802,0.326594,0.546532}
+\definecolor{color13}{rgb}{0.248629,0.278775,0.534556}
+\definecolor{color14}{rgb}{0.12156862745098,0.466666666666667,0.705882352941177}
 
 \begin{axis}[
 colorbar,
 colorbar style={ylabel={}},
-colormap={mymap}{[1pt]
-  rgb(0pt)=(0,0,0.5);
-  rgb(22pt)=(0,0,1);
-  rgb(25pt)=(0,0,1);
-  rgb(68pt)=(0,0.86,1);
-  rgb(70pt)=(0,0.9,0.967741935483871);
-  rgb(75pt)=(0.0806451612903226,1,0.887096774193548);
-  rgb(128pt)=(0.935483870967742,1,0.0322580645161291);
-  rgb(130pt)=(0.967741935483871,0.962962962962963,0);
-  rgb(132pt)=(1,0.925925925925926,0);
-  rgb(178pt)=(1,0.0740740740740741,0);
-  rgb(182pt)=(0.909090909090909,0,0);
-  rgb(200pt)=(0.5,0,0)
-},
+colormap/viridis,
 point meta max=98.5559785610705,
 point meta min=12.0628665990324,
 tick align=outside,
@@ -41,7 +42,7 @@
 .. controls (axis cs:0.5983927657594,0.52530459850483) and (axis cs:0.60872671606002,0.500356235536182) .. (axis cs:0.627118684055587,0.481964267540616)
 .. controls (axis cs:0.645510652051153,0.46357229954505) and (axis cs:0.6704590150198,0.45323834924443) .. (axis cs:0.696469185597862,0.45323834924443)
 --cycle;
-\path [fill=color0, fill opacity=0.4]
+\path [fill=color1, fill opacity=0.4]
 (axis cs:0.286139334950379,0.650985995927077)
 .. controls (axis cs:0.304301231914869,0.650985995927077) and (axis cs:0.321721709658734,0.658201794071604) .. (axis cs:0.334564110161536,0.671044194574406)
 .. controls (axis cs:0.347406510664338,0.683886595077208) and (axis cs:0.354622308808866,0.701307072821074) .. (axis cs:0.354622308808866,0.719468969785563)
@@ -52,7 +53,7 @@
 .. controls (axis cs:0.217656361091893,0.701307072821074) and (axis cs:0.224872159236421,0.683886595077208) .. (axis cs:0.237714559739223,0.671044194574406)
 .. controls (axis cs:0.250556960242025,0.658201794071604) and (axis cs:0.26797743798589,0.650985995927077) .. (axis cs:0.286139334950379,0.650985995927077)
 --cycle;
-\path [fill=color0, fill opacity=0.4]
+\path [fill=color2, fill opacity=0.4]
 (axis cs:0.226851453564203,0.375013269976025)
 .. controls (axis cs:0.239605916680458,0.375013269976025) and (axis cs:0.251839707087159,0.38008067188171) .. (axis cs:0.260858474447056,0.389099439241608)
 .. controls (axis cs:0.269877241806954,0.398118206601505) and (axis cs:0.274944643712639,0.410351997008206) .. (axis cs:0.274944643712639,0.423106460124461)
@@ -63,13 +64,13 @@
 .. controls (axis cs:0.178758263415767,0.410351997008206) and (axis cs:0.183825665321452,0.398118206601505) .. (axis cs:0.19284443268135,0.389099439241608)
 .. controls (axis cs:0.201863200041247,0.38008067188171) and (axis cs:0.214096990447948,0.375013269976025) .. (axis cs:0.226851453564203,0.375013269976025)
 --cycle;
-\path [fill=color0, fill opacity=0.4]
+\path [fill=color3, fill opacity=0.4]
 (axis cs:0,0.25)
 --(axis cs:0.707106781186548,-0.457106781186547)
 --(axis cs:1.06066017177982,-0.103553390593274)
 --(axis cs:0.353553390593274,0.603553390593274)
 --cycle;
-\path [fill=color0, fill opacity=0.4]
+\path [fill=color4, fill opacity=0.4]
 (axis cs:0.319763407924357,0.424037633715361)
 .. controls (axis cs:0.322195069846986,0.41193268399766) and (axis cs:0.32762605917823,0.400629304754155) .. (axis cs:0.335557083587924,0.39116666034644)
 .. controls (axis cs:0.343488107997617,0.381704015938725) and (axis cs:0.353668440565538,0.374381253055418) .. (axis cs:0.365162163052912,0.369871582251773)
@@ -78,14 +79,14 @@
 --(axis cs:0.392117518194151,0.438572244679624)
 --(axis cs:0.319763407924357,0.424037633715361)
 --cycle;
-\path [fill=color0, fill opacity=0.4]
+\path [fill=color5, fill opacity=0.4]
 (axis cs:0.325292534093276,0.0560527278677551)
 .. controls (axis cs:0.326029332980179,0.0524175858342907) and (axis cs:0.327857149748153,0.0490920334084667) .. (axis cs:0.330531105175883,0.0465216068731004)
 .. controls (axis cs:0.333205060603613,0.0439511803377341) and (axis cs:0.336600159513475,0.0422560353685811) .. (axis cs:0.340261548228723,0.0416632766663606)
 --(axis cs:0.343178016150869,0.0596778966095683)
 --(axis cs:0.325292534093276,0.0560527278677551)
 --cycle;
-\path [fill=color0, fill opacity=0.4]
+\path [fill=color6, fill opacity=0.4]
 (axis cs:0.717397420978084,0.384927189771934)
 .. controls (axis cs:0.719079422931577,0.383433016520297) and (axis cs:0.721033228508406,0.382276220003659) .. (axis cs:0.723152137857449,0.381519974376895)
 .. controls (axis cs:0.725271047206493,0.380763728750131) and (axis cs:0.727515773023645,0.380422055779642) .. (axis cs:0.729763727926958,0.380513614668531)
@@ -106,7 +107,7 @@
 --(axis cs:0.729049707384042,0.398044255330431)
 --(axis cs:0.717397420978084,0.384927189771934)
 --cycle;
-\path [fill=color0, fill opacity=0.4]
+\path [fill=color7, fill opacity=0.4]
 (axis cs:0.4,0.7)
 .. controls (axis cs:0.4,0.713131879680739) and (axis cs:0.397413306040759,0.726136068376074) .. (axis cs:0.392387953251129,0.738268343236509)
 .. controls (axis cs:0.387362600461498,0.750400618096944) and (axis cs:0.379996319290631,0.761425036946679) .. (axis cs:0.370710678118655,0.770710678118655)
@@ -127,7 +128,7 @@
 (axis cs:0.3,0.7)
 (axis cs:0.4,0.7)
 --cycle;
-\path [fill=color0, fill opacity=0.4]
+\path [fill=color8, fill opacity=0.4]
 (axis cs:0.9,0.8)
 .. controls (axis cs:0.9,0.826263759361477) and (axis cs:0.894826612081518,0.852272136752148) .. (axis cs:0.884775906502257,0.876536686473018)
 .. controls (axis cs:0.874725200922996,0.900801236193888) and (axis cs:0.859992638581262,0.922850073893357) .. (axis cs:0.84142135623731,0.941421356237309)
@@ -164,14 +165,14 @@
 .. controls (axis cs:0.846119959061139,0.839204102564111) and (axis cs:0.85,0.819697819521108) .. (axis cs:0.85,0.8)
 (axis cs:0.9,0.8)
 --cycle;
-\path [fill=color0, fill opacity=0.4]
+\path [fill=color9, fill opacity=0.4]
 (axis cs:1,0.3)
 .. controls (axis cs:1,0.326263759361477) and (axis cs:0.994826612081518,0.352272136752148) .. (axis cs:0.984775906502257,0.376536686473018)
 .. controls (axis cs:0.974725200922997,0.400801236193888) and (axis cs:0.959992638581262,0.422850073893357) .. (axis cs:0.94142135623731,0.441421356237309)
 --(axis cs:0.8,0.3)
 --(axis cs:1,0.3)
 --cycle;
-\path [fill=color0, fill opacity=0.4]
+\path [fill=color10, fill opacity=0.4]
 (axis cs:0.94142135623731,0.441421356237309)
 .. controls (axis cs:0.922850073893357,0.459992638581262) and (axis cs:0.900801236193888,0.474725200922997) .. (axis cs:0.876536686473018,0.484775906502257)
 .. controls (axis cs:0.852272136752148,0.494826612081518) and (axis cs:0.826263759361477,0.5) .. (axis cs:0.8,0.5)
@@ -180,23 +181,23 @@
 .. controls (axis cs:0.850400618096944,0.387362600461498) and (axis cs:0.861425036946679,0.379996319290631) .. (axis cs:0.870710678118655,0.370710678118655)
 --(axis cs:0.94142135623731,0.441421356237309)
 --cycle;
-\path [fill=color0, fill opacity=0.4]
+\path [fill=color11, fill opacity=0.4]
 (axis cs:0.722443382570222,0.322958913853178)
 --(axis cs:0.361788655622314,0.228263230878956)
 --(axis cs:0.293714046388829,0.630976123854488)
 --cycle;
-\path [fill=color0, fill opacity=0.4]
+\path [fill=color12, fill opacity=0.4]
 (axis cs:0.0921049399450752,0.433701172679528)
 --(axis cs:0.430862763329644,0.493685097650306)
 --(axis cs:0.425830290295828,0.312261222972465)
 --cycle;
-\path [fill=color0, fill opacity=0.4]
+\path [fill=color13, fill opacity=0.4]
 (axis cs:0.426351306962808,0.893389163117135)
 --(axis cs:0.94416001820388,0.501836675884337)
 --(axis cs:0.623952951792111,0.115618395079296)
 --cycle;
 
-\draw[draw=none,fill=color0,fill opacity=0.4,rotate around={45:(axis cs:1,0.5)}] (axis cs:1,0.5) ellipse (0.5 and 0.25);
+\draw[draw=none,fill=color14,fill opacity=0.4,rotate around={45:(axis cs:1,0.5)}] (axis cs:1,0.5) ellipse (0.5 and 0.25);
 \draw[draw=red,fill=red,opacity=0.4] (axis cs:0,1) circle (0.5);
 \draw[->,draw=black] (axis cs:0.25,0.25) -- (axis cs:0.5,0.25);
 \path [draw=black, -latex, draw=black]

tikzplotlib/_patch.py

@@ -67,31 +67,27 @@ def draw_patchcollection(data, obj):
     """Returns PGFPlots code for a number of patch objects.
     """
     content = []
-    # Gather the draw options.
-    try:
-        ec = obj.get_edgecolor()[0]
-    except IndexError:
-        ec = None
-
-    try:
-        fc = obj.get_facecolor()[0]
-    except IndexError:
-        fc = None
 
-    try:
-        ls = obj.get_linestyle()[0]
-    except IndexError:
-        ls = None
+    # recompute the face colors
+    obj.update_scalarmappable()
 
-    try:
-        w = obj.get_linewidth()[0]
-    except IndexError:
-        w = None
+    def ensure_list(x):
+        return [None] if len(x) == 0 else x
 
-    data, draw_options = mypath.get_draw_options(data, obj, ec, fc, ls, w)
+    ecs = ensure_list(obj.get_edgecolor())
+    fcs = ensure_list(obj.get_facecolor())
+    lss = ensure_list(obj.get_linestyle())
+    ws = ensure_list(obj.get_linewidth())
 
     paths = obj.get_paths()
-    for path in paths:
+    for i, path in enumerate(paths):
+        # Gather the draw options.
+        ec = ecs[i % len(ecs)]
+        fc = fcs[i % len(fcs)]
+        ls = lss[i % len(lss)]
+        w = ws[i % len(ws)]
+
+        data, draw_options = mypath.get_draw_options(data, obj, ec, fc, ls, w)
         data, cont, draw_options, is_area = mypath.draw_path(
             data, path, draw_options=draw_options
         )