import numpy
import matplotlib.pyplot as plt

fig_convergence = plt.figure(1,figsize=(12,6))

x = numpy.loadtxt('log_deepAI_paper_nonlin_action_long.txt')

plt.subplot(122)
plt.plot(x[:,0])
plt.xlim([0,500])
plt.ylim([-10,200])
plt.xlabel('Steps')
plt.ylabel('Free Action')
plt.axvline(x=230.0,linestyle=':')
plt.axvline(x=250.0,linestyle=':')
plt.axvline(x=270.0,linestyle=':')

ax = plt.subplot(121)
plt.plot(x[:,0])
plt.ylim([-10,200])
ax.axvspan(0, 500, alpha=0.3, color='red')
plt.xlim([0,30000])
plt.xlabel('Steps')
plt.ylabel('Free Action')

fig_convergence.subplots_adjust(left=0.07, bottom=0.1, right=0.95, top=0.95,
                    wspace=0.2, hspace=0.15)
                    
fig_convergence.savefig('fig_convergence.pdf')
                    
plt.show()

[(4, 'matplotlib.pyplot.figure', 'plt.figure', (['(1)'], {'figsize': '(12, 6)'}), True, 'import matplotlib.pyplot as plt\n'), (6, 'numpy.loadtxt', 'numpy.loadtxt', (['"""log_deepAI_paper_nonlin_action_long.txt"""'], {}), False, 'import numpy\n'), (8, 'matplotlib.pyplot.subplot', 'plt.subplot', (['(122)'], {}), True, 'import matplotlib.pyplot as plt\n'), (9, 'matplotlib.pyplot.plot', 'plt.plot', (['x[:, (0)]'], {}), True, 'import matplotlib.pyplot as plt\n'), (10, 'matplotlib.pyplot.xlim', 'plt.xlim', (['[0, 500]'], {}), True, 'import matplotlib.pyplot as plt\n'), (11, 'matplotlib.pyplot.ylim', 'plt.ylim', (['[-10, 200]'], {}), True, 'import matplotlib.pyplot as plt\n'), (12, 'matplotlib.pyplot.xlabel', 'plt.xlabel', (['"""Steps"""'], {}), True, 'import matplotlib.pyplot as plt\n'), (13, 'matplotlib.pyplot.ylabel', 'plt.ylabel', (['"""Free Action"""'], {}), True, 'import matplotlib.pyplot as plt\n'), (14, 'matplotlib.pyplot.axvline', 'plt.axvline', ([], {'x': '(230.0)', 'linestyle': '""":"""'}), True, 'import matplotlib.pyplot as plt\n'), (15, 'matplotlib.pyplot.axvline', 'plt.axvline', ([], {'x': '(250.0)', 'linestyle': '""":"""'}), True, 'import matplotlib.pyplot as plt\n'), (16, 'matplotlib.pyplot.axvline', 'plt.axvline', ([], {'x': '(270.0)', 'linestyle': '""":"""'}), True, 'import matplotlib.pyplot as plt\n'), (18, 'matplotlib.pyplot.subplot', 'plt.subplot', (['(121)'], {}), True, 'import matplotlib.pyplot as plt\n'), (19, 'matplotlib.pyplot.plot', 'plt.plot', (['x[:, (0)]'], {}), True, 'import matplotlib.pyplot as plt\n'), (20, 'matplotlib.pyplot.ylim', 'plt.ylim', (['[-10, 200]'], {}), True, 'import matplotlib.pyplot as plt\n'), (22, 'matplotlib.pyplot.xlim', 'plt.xlim', (['[0, 30000]'], {}), True, 'import matplotlib.pyplot as plt\n'), (23, 'matplotlib.pyplot.xlabel', 'plt.xlabel', (['"""Steps"""'], {}), True, 'import matplotlib.pyplot as plt\n'), (24, 'matplotlib.pyplot.ylabel', 'plt.ylabel', (['"""Free Action"""'], {}), True, 'import matplotlib.pyplot as plt\n'), (31, 'matplotlib.pyplot.show', 'plt.show', ([], {}), True, 'import matplotlib.pyplot as plt\n')]