import numpy as np
import matplotlib.pyplot as plt
from Tkinter import *
import tkMessageBox
import tkSimpleDialog

prefixo = "vetor"

def makeWidgets():
    global entries
    window = Tk()
    scrollbar = Scrollbar(window)
    scrollbar.pack( side = RIGHT, fill=Y)
    window.title('Vetores')
    form = Frame(window)
    form.pack()
    entries = {}
    numero = tkSimpleDialog.askinteger("Vetores", "Digite o numero de vetores a serem usados:")

    for ix in range(numero):
        label = "%s %s" % (prefixo , ix + 1)
        lab = Label(form, text=label)
        ent = Entry(form)
        lab.grid(row=ix, column=0)
        ent.grid(row=ix, column=1)
        entries[label] = ent

    Button(window, text="Calcular",  command=calcular).pack(side=LEFT)
    Button(window, text="Media",  command=media).pack(side=LEFT)

    window.mainloop()

def pegavalores():
    valores = []
    for chave, entrada in sorted(entries.items()):
        valores.append(entrada.get())

    return valores

def calcular():
    calcular = pegavalores()
    plt.plot(calcular)
    plt.show()

def media():
    media = pegavalores()
    elementos = len(media)
    media = np.asarray(media, dtype=np.float64)
    valormedio = np.sum(media)/elementos
    tkMessageBox.showinfo("Media", valormedio)

window = makeWidgets()