In [17]:
import matplotlib.pyplot as plt
import numpy as np
%matplotlib inline
x = np.linspace(-1, 2, 5)
y1 = 2*x + 1
y2 = x**2
plt.plot(x, y1)
plt.plot(x, y2, color = 'red', linewidth = 1, linestyle='--')
plt.xlim((-1, 2))
plt.ylim((-3, 3))
plt.xlabel('I am x')
plt.ylabel('I am y')
plt.xticks(np.linspace(-2, 2, 5))
plt.yticks([-2,-1,0,1,2], ['$sum_{i=1}^N\sin(x)$','test2','test3','test4','test5'])
#gca get current axis
ax = plt.gca()
ax.spines['right'].set_color('none')
ax.spines['top'].set_color('none')
ax.xaxis.set_ticks_position('bottom')
ax.yaxis.set_ticks_position('left')
ax.spines['bottom'].set_position(('data', 0))
ax.spines['left'].set_position(('data', 0))