sample_for_graph.py

ot = df.sample(frac=.1, random_state=1111)
ot.sort_index(inplace=True)
ot['mean'] = ot['compound'].expanding().mean()
ot['rolling'] = ot['compound'].rolling('12h').mean()

fig = plt.figure(figsize=(20,5))
ax = fig.add_subplot(111)
ax.scatter(ot['review date'],ot['compound'], label='Tweet Sentiment')
ax.plot(ot['review date'],ot['rolling'], color ='g', label='Rolling Mean')
ax.plot(ot['review date'],ot['mean'], color='r', label='Expanding Mean')
ax.set(title='Te Papa Trip Advisor Reviews over Time', xlabel='Date', ylabel='Sentiment')
ax.legend(loc='best')
fig.tight_layout()
plt.show()