![](/img/trans.png)
[英](python) Adding a colorbar by the dates of points in a 3D scatter plot with matplotlib
[英]ValueError while adding colorbar to scatter plot in python
嗨,我有一个包含多列的数据文件,并且我想添加一个带有色条的第三轴。 我想做的是将第1列绘制为y,将第5列绘制为x轴,也想在数据点上添加第4列带有颜色,但是我遇到以下错误:
ValueError: to_rgba: Invalid rgba arg "124.12834"
to_rgb: Invalid rgb arg "124.12834"
gray (string) must be in range 0-1
我的数据文件如下:
10.0 -1000.0 1300.0 2800.0 124.12834 426.450106 207.905003
10.0 -1000.0 1200.0 2000.0 127.856047 426.794984 230.104389
10.0 -500.0 1300.0 2800.0 123.074804 426.255793 205.883523
10.0 -500.0 1200.0 2000.0 127.10481 426.453441 228.650902
10.0 0.0 1300.0 2800.0 122.240729 425.953329 204.735877
10.0 0.0 1200.0 2000.0 126.435714 426.086112 227.646258
10.0 500.0 1200.0 2000.0 125.835211 425.704534 226.969429
我使用的代码是:
import numpy as np
import matplotlib.pyplot as plt
with open("test_data1.dat") as f:
data = f.read()
data = data.split('\n')
x = [row.split('\t')[5] for row in data]
y = [row.split('\t')[1] for row in data]
z = [row.split('\t')[4] for row in data]
cm = plt.cm.get_cmap('RdYlBu')
fig, ax = plt.subplots()
sc = ax.scatter(x, y, c=z, vmin=10, vmax=15,s=35.0,cmap=cm)
fig.colorbar(sc)
plt.show()
请注意,它不会在保护所有属性的情况下复制数据,因此您可以获取以下数据,这是代码中的数据字符串:
['10.0\t-1000.0\t1300.0\t2800.0\t124.12834\t426.450106\t207.905003',
'10.0\t-1000.0\t1200.0\t2000.0\t127.856047\t426.794984\t230.104389',
'10.0\t-500.0\t1300.0\t2800.0\t123.074804\t426.255793\t205.883523',
'10.0\t-500.0\t1200.0\t2000.0\t127.10481\t426.453441\t228.650902',
'10.0\t0.0\t1300.0\t2800.0\t122.240729\t425.953329\t204.735877',
'10.0\t0.0\t1200.0\t2000.0\t126.435714\t426.086112\t227.646258',
'10.0\t500.0\t1200.0\t2000.0\t125.835211\t425.704534\t226.969429']
谢谢
我删除所有split()
的'\\t'
,因为我的文本编辑器自动将'\\ t'更改为空格。
string.split(s [,sep [,maxsplit]):如果不存在可选的第二个参数sep或“无”,则单词由任意空格字符字符串(空格,制表符,换行符,返回值,换页符)分隔。
它可以正确显示该图,这是代码:
import numpy as np
import matplotlib.pyplot as plt
data = open("test_data1.dat").read().split('\n')
x = [row.split()[5] for row in data]
y = [row.split()[1] for row in data]
z = [row.split()[4] for row in data]
cm = plt.cm.get_cmap('RdYlBu')
fig, ax = plt.subplots()
sc = ax.scatter(x, y, c=z, vmin=10, vmax=15,s=35.0,cmap=cm)
fig.colorbar(sc)
plt.show()
该错误对我而言未发生,可能是您未正确分割任何行(将其打印以进行检查),或者您可以尝试重新安装matplotlib 。
您只是将值作为字符串而不是浮点数来获取。 您必须使用:
x = [float(row.split('\t')[5]) for row in data]
y = [float(row.split('\t')[1]) for row in data]
z = [float(row.split('\t')[4]) for row in data]
此外,[1]会为您提供第二行,因为python中的索引以0开头。 因此,也许您没有获得想要的列。
如果要使点具有不同的颜色,则必须调整范围vmin=10, vmax=15,
以适合您的值或将其保留为空白,然后matplotlib自动调整它们。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.