cat populations.txt

#Cargamos los módulos que vamos a necesitar
import numpy as np
from matplotlib import pyplot as plt
%matplotlib inline

#Metemos los datos del txt en un array de numpy
data = np.loadtxt('populations.txt') # Convierte data en un array bidimensional, en una matriz vamos ...
print data
year, hares, lynxes, carrots = data.T # Traspuesta
print year
print carrots

#Dibujo el gráfico
plt.plot(year,hares,    year, lynxes,     year,carrots)
plt.legend(('Hare', 'Lynx', 'Carrot'))



# Recorto la matriz de datos para quedarme solo con las poblaciones (slicing)
populations = data[:, 1:]

# Valores estadísticos de las poblaciones: Media y desviación típica
print populations.mean(axis=0)
print populations.std(axis=0)

# ¿Que especies tienen la población más alta cada año?
np.argmax(populations, axis=1)

# ¿Que años son superados los 50000 elementos por alguna población?
print populations > 50000
print ""
print populations[populations > 50000]
print ""
above_50000 = np.any(populations > 50000, axis=1) # np.any devuelve un array con True donde se cumpla alguna de las condiciones
print above_50000
print ""
print "Any above 50000: ", year[above_50000]