【发布时间】:2018-04-01 02:19:43
【问题描述】:
我正在尝试使用 5 种不同颜色显示散点图:每个“district”一个。不幸的是,我出现了这个错误:Invalid RGBA argument: nan
我的数据框如下所示:
> price surface district
> 0 1850 46 1.0
> 1 1900 41 4.0
> 2 1925 55 4.0
> 3 1960 52 2.0
> 4 2025 57 3.0
> 5 2055 60 10.0
... ... ... ...
这是我所做的:
colors =['red', 'pink', 'limegreen', 'violet', 'cyan']
color_dict = dict(zip(np.unique(house['district']),colors))
因此颜色字典看起来像这样:
{1.0: 'red', 2.0: 'pink', 3.0: 'limegreen', 4.0: 'violet', 10.0: 'cyan'}
然后我创建了一个变量 col:
col = house['district'].map(color_dict)
最后,我画了情节:
df.plot(kind = 'scatter', x = 'surface', y = 'price', c = col)
有人能解释一下这里有什么问题吗?
【问题讨论】:
-
如果你的数据框有 NaN 值,为什么不直接调用
df.dropna()? -
确实,您必须在某处具有 NaN 值。仅根据您迄今为止提供的数据,我无法复制您的问题。
标签: python matplotlib plot graph colors