我有一个相当大的数据集,包含了我用matplotlib绘制的“星粒子”的位置。这些数据被组织成列,其中三列对应于x、y和z坐标。我绘制了x和y坐标,但我希望能够使用z坐标作为颗粒的一种彩色映射。例如,拥有较大z坐标的颗粒具有较低的alpha值或着不同的颜色。当前我的代码如下:
f = open(input_file)
line = f.readline()
line = f.readline()
px=[]
py=[]
pz=[]
for line in f:
ldata = line.split()
x = np.float(ldata[2])
y = np.float(ldata[3])
z = np.float(ldata[4])
px.append(x)
py.append(y)
pz.append(z)
plt.plot(px,py,'ro',alpha=0.1)
plt.savefig(output_file)
我是 Python 的新手