In [20]:

    

import matplotlib.pyplot as plt
import numpy as np

# Definición de la función votos_por_variedad() para obtener el número de votos por cada variedad de rábano
fichero1 = open("../datos/radishsurvey.txt")
def votos_por_variedad(): 
    dicRabanos = {}
    listaRabanos = []
    for item in fichero1:
        listaSinSaltos = item.strip()
        listaTodoMayus = listaSinSaltos.title()
        listaSinEspacios = listaTodoMayus.replace('  ',' ')
        listaSinGuion = listaSinEspacios.split(' - ')
        listaRabanos.append(listaSinGuion[1])
        variedadRabano = listaSinGuion[1]
        if variedadRabano in listaSinGuion[1]:
            dicRabanos[listaSinGuion[1]] = listaRabanos.count(variedadRabano)
    return dicRabanos

# Creación del fichero donde almacenar los datos obtenidos mediante la función votos_por_variedad()
fichero2 = open("../datos/resultado.txt", "w")

dic = votos_por_variedad()
for x,y in dic.items():
    fichero2.write(x + ': ' + str(y) + '\n')
fichero2.close()


# Representación gráfica de la función votos_por_variedad()
print('\nVotos por variedad de rábano\n')

nombre = dic.keys()
votos = dic.values()                     # eje y

x = np.arange(len(votos))                # eje x
plt.bar(x, votos)                        # pinto diagrama de barras

plt.xticks(x, nombre, rotation=90) ;     # etiqueto el eje x con los nombres de las variedades


    

    




Votos por variedad de rábano







    

In [ ]: