首页 > 后端开发 > Python教程 > 如何在 Matplotlib 中创建自定义颜色图并添加色标?

如何在 Matplotlib 中创建自定义颜色图并添加色标?

Susan Sarandon
发布: 2024-11-13 05:24:02
原创
490 人浏览过

How to Create a Custom Colormap and Add a Color Scale in Matplotlib?

创建自定义颜色图并合并色标

要创建自己的颜色图,一种方法是利用 matplotlib.colors 模块中的 LinearSegmentedColormap 函数。这种方法更简单,并产生连续的色标。

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors

# Generate random data points
x, y, c = zip(*np.random.rand(30, 3) * 4 - 2)

# Define lower and upper bounds for normalization
norm = plt.Normalize(-2, 2)

# Create a list of tuples representing the values and corresponding colors
tuples = [(norm(-2.), 'red'), (norm(-1.), 'violet'), (norm(2.), 'blue')]

# Generate the colormap from the list of tuples
cmap = matplotlib.colors.LinearSegmentedColormap.from_list('', tuples)

# Plot the data points using the custom colormap
plt.scatter(x, y, c=c, cmap=cmap, norm=norm)

# Add a color scale to the plot
plt.colorbar()
plt.show()
登录后复制

此代码片段成功创建了一个从红色到紫色到蓝色平滑过渡的颜色图,范围从 -2 到 2。色标也是合并到图的右侧,可以轻松解释颜色。

以上是如何在 Matplotlib 中创建自定义颜色图并添加色标?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板