Plot a multicolored line based on a condition in Python Matplotlib


To plot a multicolored line based on a condition in Python Matplotlib, we can take the following steps −

  • Set the figure size and adjust the padding between and around the subplots.

  • Create y data points using numpy.

  • Make l and u data points to differentiate the colors.

  • Plot the u and l data points using plot() method, with different colors.

  • To display the figure, use show() method.

Example

import matplotlib.pyplot as plt
import numpy as np

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

y = np.sin(np.linspace(-10, 10, 100))

u = y.copy()
l = y.copy()

u[u <= 0] = np.nan
l[l >= 0] = np.nan

plt.plot(u, color='red')
plt.plot(l, color='blue')

plt.show()

Output

Updated on: 09-Aug-2021

3K+ Views

Kickstart Your Career

Get certified by completing the course

Get Started
Advertisements