如何将数据从python脚本发送到Matplotlib？

Question

我对编程颇为陌生，对matplotlib有疑问。我编写了一个python脚本，它从另一个程序的outfile中读入数据，然后从一列中打印出数据。

f = open('/home/student/AstroSimulation/out.0001.z0.753.AHF_halos','r') 
for line in f: 
    if line != ' ': 
     line = line.strip() # Strips end of line character 
     columns = line.split() # Splits into coloumn 
     mass = columns[8]  # Column which contains mass values 
     print(mass) 

我现在需要做的是让matplotlib以“质量”和绘图数量与平均质量打印的值。我已经阅读了matplotlib网站上的文档，但他们并没有真正解决如何从脚本获取数据（或者我没有看到它）。如果任何人都可以指出我解释我如何做到这一点的文档，那将是非常感谢。谢谢

Answer 1

你会从脚本中调用matplotlib，所以matplotlib不会“从脚本中获取数据”。你把它发送到matplotlib。

您需要保存然而，循环外的群众，但后来，它只是在它的plot()和show()函数的调用是最基本的形式：

import matplotlib.pyplot as plt 

masses = [] 

f = open('/home/student/AstroSimulation/out.0001.z0.753.AHF_halos','r') 
f.readline() # Remove header line 
for line in f: 
    if line != ' ': 
     line = line.strip() # Strips end of line character 
     columns = line.split() # Splits into coloumn 
     mass = columns[8]  # Column which contains mass values 
     masses.append(mass) 
     print(mass) 

# If neccessary, process masses in some way 

plt.plot(masses) 
plt.show() 

Answer 2

我是你，一直到“绘制平均值的总和“。也许你可以链接到一个像你想要制作的情节的图像。

在当前的脚本，您打印“大众”，要追加到列表的浮点值：

from matplotlib import pyplot 

DATAFILE = '/home/student/AstroSimulation/out.0001.z0.753.AHF_halos' 
MASS_COL = 8 

masses = [] 
with open(DATAFILE) as f: 
    f_it = iter(f)     #get an iterator for f 
    next(f_it)      #skip the first line 
    for n, line in enumerate(f_it): #now the for loop uses f_it 
     row = line.strip().split() 
     if len(row) > MASS_COL: 
      mass = row[MASS_COL] 
      try: 
       mass = float(mass) 
       masses.append(mass) 
       print "%0.3f" % mass 
      except ValueError: 
       print "Error (line %d): %s" % (n, mass) 

#crunch mass data 
plot_data = ... 

#make a plot 
pyplot.plot(plot_data) 
pyplot.show()