控制颜色顺序并在y=0处绘制一条线

2024-09-30 10:37:13 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图根据这两组数据进行绘图。除了颜色之外,这是有效的。我正在尝试匹配ufile和dfile的颜色,以便ufile[1]和dfiles[1]中的行具有相同的颜色

我怎么能得到呢

#!/usr/bin/env python3
# import np
import pylab

ufiles = ["dos_u_Co.dat",  "dos_u_Fe.dat",  "dos_u_Re.dat",  "dos_u_tot.dat"]
dfiles = ["dos_d_Co.dat",  "dos_d_Fe.dat",  "dos_d_Re.dat",  "dos_d_tot.dat"]
colors = ('b', 'g', 'r','c')


datalist = [ ( pylab.loadtxt(filename)) for filename in ufiles ]
for data in datalist:
    pylab.plot( data[:,0], data[:,1], label=datalist)

datalist = [ ( pylab.loadtxt(filename)) for filename in dfiles ]
for data in datalist:
    pylab.plot( data[:,0], -1*data[:,1], label=datalist)

pylab.show()

Tags: inimportfordata颜色filenamedatco
1条回答
网友
1楼 · 发布于 2024-09-30 10:37:13

只要在使用^{}循环通过colors时循环通过datalist,然后使用colorkwarg设置打印颜色:

import pylab

ufiles = ["dos_u_Co.dat",  "dos_u_Fe.dat",  "dos_u_Re.dat",  "dos_u_tot.dat"]
dfiles = ["dos_d_Co.dat",  "dos_d_Fe.dat",  "dos_d_Re.dat",  "dos_d_tot.dat"]
colors = ('b', 'g', 'r','c')


datalist = [ ( pylab.loadtxt(filename)) for filename in ufiles ]
for data, color in zip(datalist, colors):
    pylab.plot( data[:,0], data[:,1], color=color, label=datalist)

datalist = [ ( pylab.loadtxt(filename)) for filename in dfiles ]
for data, color in zip(datalist, colors):
    pylab.plot( data[:,0], -1*data[:,1], color=color, label=datalist)

pylab.show()

相关问题 更多 >

    热门问题