scatter绘制三维图形报错:ValueError: Invalid RGBA argument

本文解决在使用matplotlib的scatter函数绘制三维图形时遇到的ValueError错误,原因是输入数据包含复数,而非实数。通过使用numpy的real函数将复数转换为实数,成功解决了绘图问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

scatter绘制三维图形报错:ValueError: Invalid RGBA argument

代码如下

    fig = plt.figure() 
    ax = plt.axes(projection='3d')
    c1=list(np.reshape(Y_train,(50,)))
    ax.scatter(X_train[:,2],X_train[:,1], X_train[:,0],c=c1)  #报错代码

    plt.show() 

其中X_train为
X_train的类型

报错内容如下:

masked_array(data=[0.267004  +0.j, 0.004874  +0.j, 0.329415  +0.j,
                   0.50135356+0.j],
             mask=False,
       fill_value=(1e+20+0j))

<Figure size 432x288 with 1 Axes>

找了其他的博客,了解到scatter的第三个参数需要为数值型,即X_train[:,2]这一维度需要为实数,而这里是复数。
使用np.real()修改报错代码即可:

 ax.scatter(X_train[:,2],X_train[:,1], np.real(X_train[:,0]),c=c1) 
import pandas as pd import matplotlib.pyplot as plt data = pd.DataFrame({ '物种名称': ['熊猫', '狗', '兔子', '乌龟', '鬣狗', '企鹅', '蛇', '鸭子', '马', '鲨鱼'], '体长': [100, 60, 40, 50, 120, 70, 80, 60, 220, 400], '体重': [100, 30, 3, 20, 30, 40, 4, 3, 500, 700], '速度': [32, 56, 72, 5, 70, 10, 10, 16, 88, 45], '分类类型': ['哺乳动物', '哺乳动物', '哺乳动物', '爬行动物', '哺乳动物', '鸟类', '爬行动物', '鸟类', '哺乳动物', '鱼类']}) # 定义2行2列的图形 fig, axs = plt.subplots(nrows=2, ncols=2, figsize=(12, 8)) # 1行1列的子图:横向柱形图 axs[0, 0].barh(data['物种名称'], data['体长'], color=data['分类类型']) axs[0, 0].set_xlabel('体长') axs[0, 0].set_ylabel('物种名称') axs[0, 0].set_title('各物种体长横向柱形图') # 1行2列的子图:折线图 axs[0, 1].plot(data['物种名称'], data['体重'], '-o', color=data['分类类型']) axs[0, 1].set_xlabel('物种名称') axs[0, 1].set_ylabel('体重') axs[0, 1].set_title('各物种体重折线图') # 2行1列的子图:散点图 axs[1, 0].scatter(data['物种名称'], data['速度'], c=data['分类类型']) axs[1, 0].set_xlabel('物种名称') axs[1, 0].set_ylabel('速度') axs[1, 0].set_title('各物种速度散点图') # 2行2列的子图:饼图 grouped_data = data.groupby('分类类型').size() axs[1, 1].pie(grouped_data, labels=grouped_data.index, autopct='%1.1f%%') axs[1, 1].set_title('各分类类型占比饼图') plt.tight_layout() plt.show()此代码报错为Traceback (most recent call last): File "C:/Users/lenovo/OneDrive/桌面/绘图/绘图1.py", line 38, in <module> axs[0, 0].barh(data['物种名称'], data['体长'], color=data['分类类型']) File "D:\py\Lib\site-packages\matplotlib\axes\_axes.py", line 2649, in barh patches = self.bar(x=left, height=height, width=width, bottom=y, File "D:\py\Lib\site-packages\matplotlib\__init__.py", line 1459, in inner return func(ax, *map(sanitize_sequence, args), **kwargs) File "D:\py\Lib\site-packages\matplotlib\axes\_axes.py", line 2441, in bar color = itertools.chain(itertools.cycle(mcolors.to_rgba_array(color)), File "D:\py\Lib\site-packages\matplotlib\colors.py", line 487, in to_rgba_array rgba = np.array([to_rgba(cc) for cc in c]) File "D:\py\Lib\site-packages\matplotlib\colors.py", line 487, in <listcomp> rgba = np.array([to_rgba(cc) for cc in c]) File "D:\py\Lib\site-packages\matplotlib\colors.py", line 299, in to_rgba rgba = _to_rgba_no_colorcycle(c, alpha) File "D:\py\Lib\site-packages\matplotlib\colors.py", line 374, in _to_rgba_no_colorcycle raise ValueError(f"Invalid RGBA argument: {orig_c!r}") ValueError: Invalid RGBA argument: '哺乳动物'解释错误原因并给出正确代码
06-07
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值