Warm tip: This article is reproduced from stackoverflow.com, please click
matplotlib python

How to plot a gradient color line?

发布于 2021-01-30 00:40:13

This code plots a scatterplot with gradient colors:

import matplotlib.pyplot as plt
import numpy as np

x = np.arange(30)
y = x
t = x

plt.scatter(x, y, c=t)
plt.colorbar()
plt.show()

enter image description here

but how would I plot a gradient color line with a x and y coordinate?

Questioner
Artur Müller Romanov
Viewed
0
分享
Alperino 2020-10-09 01:50

Did you already have a look at https://matplotlib.org/3.1.1/gallery/lines_bars_and_markers/multicolored_line.html?

EDIT: as suggested in the comments, a minimal working example could be

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

x    = np.linspace(0,1, 100)
y    = np.linspace(0,1, 100)
cols = np.linspace(0,1,len(x))

points = np.array([x, y]).T.reshape(-1, 1, 2)
segments = np.concatenate([points[:-1], points[1:]], axis=1)

fig, ax = plt.subplots()
lc = LineCollection(segments, cmap='viridis')
lc.set_array(cols)
lc.set_linewidth(2)
line = ax.add_collection(lc)
fig.colorbar(line,ax=ax)

enter image description here

Artur Müller Romanov 1970-01-01 08:00:00

yes but I can't seem to get it to work on a simple line

Alperino 1970-01-01 08:00:00

Simple lines in matplotlib, i.e. Line2d-objects, only support one color at once. If you want to have a color gradient, you will need to work with LineCollections.

Quang Hoang 1970-01-01 08:00:00

The technique is to mesh the line into small pieces and plot each with a color. The provided link shows how it works indeed.

Taslim Oseni 1970-01-01 08:00:00

While this link may answer the question, it is better to include the essential parts of the answer here and provide the link for reference. Link-only answers can become invalid if the linked page changes. - From Review

Related issues

1
How to use python cut method to create bins, accept one parameter and return appropriate bin?
2
Create a dictionary from a list of lists with certain criteria
3
selecting columns based on row value, Python, Pandas
4
plotting count of zeros and ones in a dataframe
5
BeautifulSoup find.all() web scraping returns empty
6
python function. output a keys list from a dictionary if the key is todays date
7
Best way to perform multiple amount of Pandas lookups between two DataFrames
8
How to get the number of columns and the width of each column in a Pandas pivot table?
9
Display a column when a desired value is missing while grouping in Pandas dataframe
10
Python hide ticks but show tick labels