如何使用 Matplotlib 在 Python 中为分类级别创建具有不同颜色的散点图?

Mary-Kate Olsen
发布: 2024-10-17 16:32:02
原创
851 人浏览过

How to Create Scatter Plots with Distinct Colors for Categorical Levels in Python Using Matplotlib?

使用 Matplotlib 在 Python 中绘制不同颜色的分类级别散点图

在 Matplotlib 中,一个用于创建静态、动画和动画的 Python 库在 Python 中的交互式可视化中,您可以利用 plt.scatter 函数的 c 参数,为分类变量的每个级别绘制不同颜色的不同散点图。

<code class="python">import matplotlib.pyplot as plt

df = pd.DataFrame({'x': [1, 2, 3], 'y': [4, 5, 6], 'color': ['red', 'blue', 'green']})

colors = {'red': 'tab:red', 'blue': 'tab:blue', 'green': 'tab:green'}

plt.scatter(df['x'], df['y'], c=df['color'].map(colors))
plt.show()</code>
登录后复制

通过传递 c 参数,一个字典将颜色名称映射到 RGB 值可用于指定每个点的颜色。然后,Pandas 的 map 方法将颜色映射应用到 df['color'] 列,有效地为每个点分配唯一的颜色。

<code class="python">colors = {'D': 'tab:blue', 'E': 'tab:orange', 'F': 'tab:green', 'G': 'tab:red', 'H': 'tab:purple', 'I': 'tab:brown', 'J': 'tab:pink'}

ax.scatter(df['carat'], df['price'], c=df['color'].map(colors))</code>
登录后复制

这种方法允许更自定义的配色方案并更好地控制绘图中使用的颜色。通过使用颜色字典,用户可以根据需要轻松修改配色方案。

以上是如何使用 Matplotlib 在 Python 中为分类级别创建具有不同颜色的散点图?的详细内容。更多信息请关注PHP中文网其他相关文章!

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