Matplotlib.pyplot.sca() in Python

# Implementation of matplotlib function
import matplotlib.pyplot as plt
from scipy import sin, cos


fig, ax = plt.subplots(2, 1)
x = [1, 2, 3, 4, 5, 6, 7, 8, 9]

y1 = sin(x)
y2 = cos(x)

plt.sca(ax[0])
plt.plot(x, y1)

plt.sca(ax[1])
plt.plot(x, y2)

fig.suptitle("""matplotlib.pyplot.sca()
function Example\n\n""", fontweight ="bold") 
  
plt.show() 

# Implementation of matplotlib function
import matplotlib.pyplot as plt
 

fig, axes = plt.subplots(2, 2) 
axes = axes.flatten() 

for i in range(4):
    plt.sca(axes[i])
    axes[i].text(0.5, 0.5, i + 1)
    
fig.suptitle("""matplotlib.pyplot.sca()
function Example\n\n""", fontweight ="bold") 
  
plt.show()