Come stampare i dati da più file di testo a due colonne con legende in Matplotlib?

Question

Come posso aprire più file di testo da diverse directory e tracciarli su un singolo grafico con legende?

Answer 1

Assumere il file assomiglia a questo ed è denominato test.txt (spazio delimitato):

1 2 
3 4 
5 6 
7 8 

Poi:

#!/usr/bin/python 

import numpy as np 
import matplotlib.pyplot as plt 

with open("test.txt") as f: 
    data = f.read() 

data = data.split('\n') 

x = [row.split(' ')[0] for row in data] 
y = [row.split(' ')[1] for row in data] 

fig = plt.figure() 

ax1 = fig.add_subplot(111) 

ax1.set_title("Plot title...")  
ax1.set_xlabel('your x label..') 
ax1.set_ylabel('your y label...') 

ax1.plot(x,y, c='r', label='the data') 

leg = ax1.legend() 

plt.show() 

trovo che la consultazione del sito gallery of plots on the matplotlib utile per capire le etichette delle leggende e degli assi.

Answer 2

Ciò è relativamente semplice se si utilizza direttamente pylab (incluso in matplotlib) anziché matplotlib. Inizia con un elenco di nomi di file e nomi di legenda, come [('nome del file 1', 'etichetta 1'), ('nome del file 2', 'etichetta 2'), ...]. Quindi è possibile utilizzare qualcosa come il seguente:

import pylab 

datalist = [ (pylab.loadtxt(filename), label) for filename, label in list_of_files ] 

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

pylab.legend() 
pylab.title("Title of Plot") 
pylab.xlabel("X Axis Label") 
pylab.ylabel("Y Axis Label") 

È inoltre possibile aggiungere qualcosa come fmt = 'o' al comando trama, al fine di cambiare da una linea a punti. Per impostazione predefinita, matplotlib con pylab traccia la stessa figura senza cancellarla, quindi è possibile eseguire il comando di stampa più volte.

Answer 3

mi sento il modo più semplice sarebbe

from matplotlib import pyplot; 
from pylab import genfromtxt; 
mat0 = genfromtxt("data0.txt"); 
mat1 = genfromtxt("data1.txt"); 
pyplot.plot(mat0[:,0], mat0[:,1], label = "data0"); 
pyplot.plot(mat1[:,0], mat1[:,1], label = "data1"); 
pyplot.legend(); 
pyplot.show(); 

1. etichetta è la stringa visualizzata sulla leggenda
2. si può tracciare come molte serie di punti dati possibile prima di show() per tracciare tutti sullo stesso grafico Questo è il modo semplice per tracciare grafici semplici. Per altre opzioni in genfromtxt vai a questo url.