Python - Ayuda con gráficas en clases.

#-*- coding: utf-8 -*-

import numpy as np

import matplotlib.pyplot as plt

from numpy import sin,cos,tan,log,exp,log10

class Graficos(object):

	"""

	fx  -  Función f(x)

	a   -  Extremo inferior del intervalo

	b   -  Extremo superior del intervalo

	n   -  Número de puntos a graficar

	"""

	def __init__(self,fx,a,b,n):

		self.X = np.linspace(a, b, n)

		x = self.X # Aux var --> Para evaluar f(x)

		self.Y = eval(fx)

	def linea(self):

		self.fig = plt.figure()

		self.ax = self.fig.add_subplot(111)

		self.ax.plot(self.X, self.Y)

		plt.show()

	def barras(self):

		self.fig = plt.figure()

		self.ax = self.fig.add_subplot(111)

		self.ax.bar(self.X, self.Y, align="center")

		plt.show()

	def puntos(self):

		self.fig = plt.figure()

		self.ax = self.fig.add_subplot(111)

		self.ax.scatter(self.X, self.Y)

		plt.show()

def main():

	fx = raw_input("f(x) = ")

	a = input("a = ")

	b = input("b = ")

	n = input("n = ")

	gr = Graficos(fx, a, b, n)

	menu = u"""

	Seleccione el tipo de gráfica:

	1. Linea

	2. Barras

	3. Puntos

	Presione cualquier otra opción para salir

	"""

	while True:

		print menu

		opc = input('')

		if opc == 1:

			gr.linea()

		elif opc == 2:

			gr.barras()

		elif opc == 3:

			gr.puntos()

		else:

			break

if __name__=='__main__':

	main()