您可以使用 Matplotlib 根据 Pandas 中的特定列值绘制彩色散点图吗？

使用 Pandas 和 Matplotlib 按列值对散点图着色

Matplotlib 是一个流行的 Python 库，用于在Python。本文探讨使用 Matplotlib 根据 Pandas DataFrame 特定列中的值对散​​点图进行着色。

导入和数据

首先，我们导入必要的库，包括 Matplotlib（作为 plt）和 Pandas（作为 pd）。我们还生成一个包含三列的示例 DataFrame (“df”)：“身高”、“体重”和“性别”。

<code class="python">import matplotlib.pyplot as plt
import pandas as pd
import numpy as np

np.random.seed(0)
N = 37
_genders = ["Female", "Male", "Non-binary", "No Response"]
df = pd.DataFrame({
    "Height (cm)": np.random.uniform(low=130, high=200, size=N),
    "Weight (kg)": np.random.uniform(low=30, high=100, size=N),
    "Gender": np.random.choice(_genders, size=N),
})</code>

2021 年 8 月更新

Seaborn引入了新的图形级函数，例如0.11.0版本中的seaborn.relplot。推荐使用这些函数而不是直接使用 FacetGrid。

<code class="python">sns.relplot(data=df, x="Weight (kg)", y="Height (cm)", hue="Gender", hue_order=_genders, aspect=1.61)
plt.show()</code>

旧答案 (2015)

如果你想直接使用 Matplotlib，你需要映射 matplotlib将函数分散到 Pandas DataFrame 的类别上。为此：

• 根据列和颜色创建一个包含唯一类别的字典。
• 向 DataFrame 添加一个新的“颜色”列，为每个类别分配相应的颜色。
• 使用散点函数绘制数据，指定颜色列作为“c”参数。

<code class="python">def dfScatter(df, xcol='Height', ycol='Weight', catcol='Gender'):
    fig, ax = plt.subplots()
    categories = np.unique(df[catcol])
    colors = np.linspace(0, 1, len(categories))
    colordict = dict(zip(categories, colors))

    df["Color"] = df[catcol].apply(lambda x: colordict[x])
    ax.scatter(df[xcol], df[ycol], c=df.Color)
    return fig

fig = dfScatter(df)
fig.savefig('fig1.png')</code>

按照以下步骤，您可以轻松地根据颜色绘制散点图使用 Pandas 和 Matplotlib 的列值。

以上是您可以使用 Matplotlib 根据 Pandas 中的特定列值绘制彩色散点图吗？的详细内容。更多信息请关注PHP中文网其他相关文章！