import matplotlib.pyplot as plt
import numpy as np

# Datos
afirmaciones = ['Nerviosismo', 'Preocupación', 'Evitación', 'Abrumado', 'Estrés']
porcentajes = [
    [0, 37.5, 18.75, 25, 18.75],   # Nerviosismo
    [6.25, 25, 18.75, 31.25, 18.75], # Preocupación
    [12.5, 31.25, 12.5, 37.5, 6.25], # Evitación
    [0, 18.75, 31.25, 37.5, 12.5],   # Abrumado
    [6.25, 12.5, 31.25, 37.5, 12.5]   # Estrés
]

# Crear gráfico
labels = ['Muy en Desacuerdo', 'En Desacuerdo', 'Neutral', 'De Acuerdo', 'Muy de Acuerdo']
bar_width = 0.35
x = np.arange(len(afirmaciones))

# Crear gráfico apilado
for i in range(len(labels)):
    plt.bar(x, [p[i] for p in porcentajes], bar_width if i ==0 else bar_width/2,
            label=labels[i], bottom=np.sum(porcentajes[:i], axis=0))

plt.xlabel('Afirmaciones')
plt.ylabel('Porcentaje (%)')
plt.title('Distribución Porcentual de Ansiedad Matemática')
plt.xticks(x, afirmaciones)
plt.legend(title='Respuestas')
plt.show()