nile.py

import copent
import matplotlib.pyplot as plt
import numpy as np

nile = [1120,1160,963,1210,1160,1160,813,1230,1370,1140,995,935,1110,
994,1020,960,1180,799,958,1140,1100,1210,1150,1250,1260,1220,1030,1100,
774,840,874,694,940,833,701,916,692,1020,1050,969,831,726,456,824,702,
1120,1100,832,764,821,768,845,864,862,698,845,744,796,1040,759,781,865,
845,944,984,897,822,1010,771,676,649,846,812,742,801,1040,860,874,848,890,
744,749,838,1050,918,986,797,923,975,815,1020,906,901,1170,912,746,919,718,
714,740]
year = np.linspace(1871,1970,100)

pos,maxstat,stat = copent.cpd(nile)
print(pos)

plt.subplot(2,1,1)
plt.plot(year,nile)
m1 = np.mean(nile[0:(pos-1)])
m2 = np.mean(nile[pos:])
pos = pos + 1870
plt.plot([1871,pos],[m1,m1], linestyle = '--', color = 'r')
plt.plot([pos+1,len(nile)+1871],[m2,m2], linestyle = '--', color = 'r')
plt.subplot(2,1,2)
plt.plot(year,stat)
plt.plot([pos,pos],[min(stat),max(stat)],linestyle = '--', color = 'r')
plt.show()