ax1_2 = ax1.twinx()
ax1_2.ticklabel_format(axis='y', style='sci', scilimits=(0, 1))
ax1_2.set_ylabel('Potential', c='tab:green')
ax1_2.tick_params(axis='y', labelcolor='tab:green')
ax2.set_ylabel('Concentration')
# for l= 10e-6
ax1.set_ylim([-4e6,4e6])
ax1_2.set_ylim(-8,8)
ax2.set_ylim([0, 2e23])
ax3.set_ylim([0,2e23])
ax3.set_ylabel("Nion_1 at x=0")
ax4.set_ylabel("Potential_1 at x=0")
ax1.axhline(y=0)
Efield = -potential.grad()[0]
l1 = ax1.plot(x, Efield, c='grey')
l2 = ax1_2.plot(x, potential.value, c='tab:green')
l3 = ax2.plot(x, Pion.value,label='Positive_1', c='tab:red')
l4 = ax2.plot(x, Nion.value,label='Negative_1', c='tab:blue')
# l5 = ax3.plot(x, Hole.value, label='Positive_2', c='red')
# l6 = ax3.plot(x, Electron.value, label='Negative_2', c='green')
# ax3.plot(x, Electron.value,label='Electron', c='tab:blue')
# ax3.plot(x, Hole.value,label="Hole", c='tab:red')
ax3.scatter(step, Nion.value[0], c='r', s=1)
# ax4.scatter(step, Hole.value[0], c='b', s=1)
Efield_save[:, step] = Efield
ax2.set_ylabel('Concentration')
ax3.set_ylabel("Pion at x=0")
# for l= 1e-6
ax1.set_ylim([-1e5, 1e5])
ax1_2.set_ylim(-0.02, 0.02)
ax2.set_ylim([-1e10, 2e21])
#
# for l=10e-6
# ax1.set_ylim([-1e6, 1e6])
# ax1_2.set_ylim(-0.5, 0.5)
# ax2.set_ylim([-1e10, 7e21])
# ax3.set_xlim(0, 1000)
l1 = ax1.plot(x, potential.grad()[0], c='grey')
l2 = ax1_2.plot(x, potential.value, c='tab:green')
l3 = ax2.plot(x, Pion.value, label='Positive', c='tab:red')
l4 = ax2.plot(x, Nion.value, label='Negative', c='tab:blue')
ax3.scatter(step, Pion.value[0], c='r', s=1)
# potential_grad_save[:, step] = potential.grad()[0]
# potential_save[:, step] = potential.value
# Pion_save[:, step] = Pion.value
# Nion_save[:, step] = Nion.value
ax2.legend(loc="upper right")
fig.suptitle('Step: ' + str(step))
fig.show()
