import torch
from torch import nn
from IPython.display import clear_output
import numpy as np
import matplotlib.pyplot as plt
import math
from math import pi
sampleX=torch.linspace(0,13,14)
sampleY=torch.zeros_like(sampleX)
for i in range(14):
    if(int(sampleX[i])%2==0):sampleY[i]=0.3
    else:sampleY[i]=0.5
    if(9<=sampleX[i] or sampleX[i]<0):
        sampleY[i]*=0.9
        sampleY[i]+=0.2
sampleY*=100
testX=sampleX
testY=sampleY
img=plt.figure(figsize=(6.5,3))
xlim=(0,14)
ylim=(20,70)
origin=(0,20)
plt.xlim(xlim)
plt.ylim(ylim)
ax=plt.gca()
ax.spines["left"].set_position(("data",origin[0]))
ax.spines["bottom"].set_position(("data",origin[1]))
ax.spines['right'].set_color("white")
ax.spines['top'].set_color("white")
ax.xaxis.set_ticks_position("bottom")
ax.yaxis.set_ticks_position("left")
x=np.linspace(0,13,14)+0.5
ticks=[]
for i in range(14):
    ticks.append(["Mon","Tue","Wed","Thu","Fri","Sat","Sun"][int(i/2)])
    if(i%2==0):ticks[i]+="\nMorn"
    else:ticks[i]+="\nEve"
plt.xticks(x,ticks,fontsize=12)
plt.annotate(s="",xy=(xlim[1]+0.0005*(xlim[1]-xlim[0]),origin[1]),xycoords="data",xytext=(-0.0001,0),textcoords="offset points",fontsize=12,
            arrowprops=dict(arrowstyle='->'))
plt.annotate(s="",xy=(origin[0],ylim[1]+0.0005*(ylim[1]-ylim[0])),xycoords="data",xytext=(0,-0.0001),textcoords="offset points",fontsize=12,
            arrowprops=dict(arrowstyle='->'))
ax.text(x=origin[0]+0.01*(xlim[1]-origin[0]),y=ylim[1]-0.05*(ylim[1]-origin[1]),s=r"$\alpha($%$)$",fontsize=13)
plt.bar(np.arange(14)+0.5,sampleY,facecolor="#9999ff",edgecolor="white")
#plt.plot(sampleX,sampleY)
plt.show()