文章目录

  • 前言
  • 一、使用字符串颜色：
  • 二、使用十六进制颜色：
  • 三、使用RGB元组：
  • 四、使用颜色映射：
  • 总结

前言

在matplotlib中，plt.plot()函数可以接受颜色参数，可以设置为字符串颜色（如'red'），也可以设置为十六进制颜色（如’#FF0000’表示红色），也可以设置为RGB元组（如(1,0,0)表示红色）。

如果想要用数值（如0，1）来设置颜色，可以使用颜色映射（Colormap）。

在matplotlib中，'viridis'，'plasma'，'inferno'，'magma'和'cividis'等都是预定义的颜色映射。

一、使用字符串颜色：

import matplotlib.pyplot as pltimport numpy as npx = np.linspace(0, 10, 100)y = np.sin(x)plt.plot(x, y, color='red')# 使用红色画线plt.show()

二、使用十六进制颜色：

import matplotlib.pyplot as pltimport numpy as npx = np.linspace(0, 10, 100)y = np.sin(x)plt.plot(x, y, color='#FF0000')# 使用红色画线plt.show()

三、使用RGB元组：

import matplotlib.pyplot as pltimport numpy as npx = np.linspace(0, 10, 100)y = np.sin(x)plt.plot(x, y, color=(1, 0, 0))# 使用红色画线plt.show()

四、使用颜色映射：

在这个例子中，我们将使用’viridis’颜色映射，将0映射为一种颜色，1映射为另一种颜色。首先，我们需要创建一个从0到1的数组，然后将其映射到颜色空间。然后，我们将这个颜色数组传递给plot()函数。

import matplotlib.pyplot as pltimport numpy as npfrom matplotlib.cm import get_cmapx = np.linspace(0, 10, 100)y = np.sin(x)z = np.linspace(0, 1, 100)# 创建一个从0到1的数组color_map = get_cmap('viridis')# 获取颜色映射colors = color_map(z)# 将数组映射到颜色空间plt.plot(x, y, color=colors)# 使用颜色数组画线plt.show()

总结

在Python的matplotlib库中，plt.plot()函数是用于绘制2D图形的常用函数。plt.plot()函数可以接受多个参数，其中包括用于设置线条颜色的参数。