/TypeScript/Multicolor points in one legend entry python

Multicolor points in one legend entry python

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.legend_handler import HandlerLineCollection
from matplotlib.collections import LineCollection

class HandlerColorLineCollection(HandlerLineCollection):
    def create_artists(self, legend, artist ,xdescent, ydescent,
                        width, height, fontsize,trans):
        x = np.linspace(0,width,self.get_numpoints(legend)+1)
        y = np.zeros(self.get_numpoints(legend)+1)+height/2.-ydescent
        points = np.array([x, y]).T.reshape(-1, 1, 2)
        segments = np.concatenate([points[:-1], points[1:]], axis=1)
        lc = LineCollection(segments, cmap=artist.cmap,
                     transform=trans)
        lc.set_array(x)
        lc.set_linewidth(artist.get_linewidth())
        return [lc]

t = np.linspace(0, 10, 200)
x = np.cos(np.pi * t)
y = np.sin(t)
points = np.array([x, y]).T.reshape(-1, 1, 2)
segments = np.concatenate([points[:-1], points[1:]], axis=1)

lc = LineCollection(segments, cmap=plt.get_cmap('copper'),
                    norm=plt.Normalize(0, 10), linewidth=3)
lc.set_array(t)

fig, ax = plt.subplots()
ax.add_collection(lc)

plt.legend([lc], ["test"],\
    handler_map={lc: HandlerColorLineCollection(numpoints=4)}, framealpha=1)

ax.autoscale_view()
plt.show()

Source: stackoverflow.com

Similar codes

1 vote350 views1 answer
React filter contains
TypeScript
1 vote315 views1 answer
Type 'MutableRefObject<undefined>' is not assignable to type 'RefObject<HTMLInputElement>'. Types of property 'current' are incompatible. Type 'undefined' is not assignable to type 'HTMLInputElement | null'. TS2322
TypeScript
8 votes332 views2 answers
Remove duplicates objects from array angular
TypeScript
2 votes335 views2 answers
File_get_contents follow redirect
TypeScript
24 votes328 views4 answers
NgFor
TypeScript
2 votes318 views1 answer
Angular send mailto html
TypeScript
0 vote303 views1 answer
Best esports game ever
TypeScript
5 votes301 views2 answers
Response.json results in pretty data python
TypeScript
34 votes359 views3 answers
Serverless.ps1 cannot be loaded because running scripts is disabled on this system.
TypeScript
0 vote362 views1 answer
How to set date axes limits in matplotlib plot
TypeScript