gpt4 book ai didi

python - 绘制 pandas DataFrame 中的数据,点的颜色取决于列

转载 作者:行者123 更新时间:2023-11-30 22:17:25 25 4
gpt4 key购买 nike

我有一个包含 3 列的 pandas DataFrame,如下所示。

col1 值标志
1 0 0
2 0.03915 0
3 0.13 1

我想从此数据帧创建一个散点图,其中 col1 是 x 轴,value 是 y 轴,我正在努力做的部分是,对于具有 flag=0 的行,我希望该点的颜色为蓝色,同样,如果 flag=1 我希望将该点着色为红色。

是否有一个简单的方法可以检查每行的标志列并相应地为点着色?

最佳答案

您可以使用内置的 df.plot.scatter() 函数并将 flag 列传递给 color 参数:

import pandas as pd

idx = [1,2,3]
value = [0., 0.03, 0.13]
flag = [0, 0, 1]

df = pd.DataFrame(dict(idx=idx, value=value, flag=flag))

df.plot.scatter('idx', 'value', c='flag', cmap='RdBu_r')

enter image description here

关于python - 绘制 pandas DataFrame 中的数据,点的颜色取决于列,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49684037/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com