python使用matplotlib画彩虹色散点图_嘻皮悠

个人资料

微博

正文字体大小：大中小

python使用matplotlib画彩虹色散点图

(2019-05-18 03:16:31)

标签：

import sys

import matplotlib as mpl

mpl.use('Agg')

import matplotlib.pyplot as plt

cm = plt.cm.get_cmap('Spectral') ＃彩虹色，其他颜色详见： https://python-graph-gallery.com/python-colors/

f=open(sys.argv[1],"r")

xx=[]

yy=[]

col=[]

for x in f:

x=x.strip()

info=x.split()

xx.append(float(info[0]))

yy.append(float(info[1]))

col.append(float(info[2]))

fig = plt.figure(figsize=(8,6))

plt.scatter(xx,yy,marker=".",facecolors='none',c=col,vmin=min(col),vmax=max(col),cmap=cm,s=1)

plt.colorbar(ticks=list(set(col))) ＃set(col) 用于unique col list; 颜色过于稠密，可选择不加xticks参数

plt.grid(True)

fig.savefig(sys.argv[2],format='png',dpi=300)

阅读┊ 收藏 ┊ 喜欢 ▼ ┊打印┊举报/Report