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

Question

我对编程相当陌生，有一个关于 matplotlib 的问题。我编写了一个 python 脚本，它从另一个程序的输出文件中读取数据，然后打印出一列中的数据。

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()