Как я могу отправить данные из скрипта Python в Matplotlib? - PullRequest
Новая
       10

Как я могу отправить данные из скрипта Python в Matplotlib?

1 голос
/ 24 августа 2011

Я довольно новичок в программировании и у меня есть вопрос, касающийся 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, но на самом деле они не обращаются к тому, как получить данные из скрипта (или я просто не видел их).Если кто-то может указать мне какую-то документацию, которая объясняет, как я это делаю, это будет очень признательно.Спасибо

Ответы [ 2 ]

2 голосов
/ 24 августа 2011

Вы будете вызывать 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()
1 голос
/ 24 августа 2011

Я был с вами вплоть до "наметить сумму над средним". Возможно, вы могли бы сделать ссылку на изображение, похожее на сюжет, который вы хотите сделать.

В вашем текущем скрипте, где вы печатаете 'mass', вы хотите добавить в список значение с плавающей запятой: 

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