ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

Python 3d scatterplot colormap问题

2019-05-29 16:53:29  阅读:246  来源: 互联网

标签:python matplotlib color-mapping


我有四维数据(x,y,z位移;和各自的电压),我希望在python中的3d散点图中绘制.我已经获得了渲染的3d图,但我希望使用颜色图来改变点的颜色,这取决于点电压的大小.

我已经尝试了一些东西,但似乎无法让它工作我收到错误ValueError:无法转换参数类型< type'numpy.ndarray'>到rgba数组.我不确定如何转换我需要转换的内容,所以如果有人可以提供一些帮助,我会非常感激.

我的代码在这里:

fig = plt.figure()
from mpl_toolkits.mplot3d import Axes3D
cmhot = plt.cm.get_cmap("hot")
ax = fig.add_subplot(111, projection='3d',)
ax.scatter(x, y, z, v, s=50, c = cmhot)
plt.show()

解决方法:

ax.scatter可以采用颜色参数c,其是标量的序列(例如列表或数组),以及cmap参数以指定颜色图.因此,要使颜色根据电压的大小而变化,您可以定义:

c = np.abs(v)

这使得正电压和负电压具有相同的颜色.相反,如果你希望每种颜色(正面或负面)都有自己的颜色,你可以使用c = v.

例如,

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
x, y, z, v = (np.random.random((4,100))-0.5)*15
c = np.abs(v)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
cmhot = plt.get_cmap("hot")
cax = ax.scatter(x, y, z, v, s=50, c=c, cmap=cmhot)

plt.show()

标签:python,matplotlib,color-mapping
来源: https://codeday.me/bug/20190529/1179470.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有