[pandas-snippet] pandas snippets #python #pandas #datascience

data-reader.py

from pandas_datareader import data
goog = data.DataReader('GOOG', start='2014', end='2020', data_source='yahoo')
goog.columns = goog.columns.str.lower()
goog = goog.asfreq('D', method='bfill')

pandas-draw-with-chinese.py

from matplotlib import rcParams
from matplotlib.font_manager import findfont, FontProperties

rcParams['font.family'] = 'sans-serif'
rcParams['font.sans-serif'] = [
    u'Microsoft Yahei',
    u'Heiti SC',
    u'Heiti TC',
    u'STHeiti',
    u'WenQuanYi Zen Hei',
    u'WenQuanYi Micro Hei',
    u"文泉驿微米黑",
    u'SimHei',
]
rcParams['axes.unicode_minus'] = False

resample-asfreq.py

goog.close.plot(alpha=0.5, style='-', figsize=(12, 6));
goog.close.asfreq('BA').plot(style='--o');
goog.close.resample('BA').mean().plot(style=':o');
plt.legend(['input', 'asfreq', 'resample'], loc='upper left');

rolling-mean.py

rolling = goog.close.rolling(365, center=True)
data = pd.DataFrame({
    'input': goog.close,
    'one-year rolling-mean': rolling.mean(),
    'one-year rolling-std': rolling.std()
})

ax = data.plot(style=['-', '--', ':'], figsize=(12, 6));
ax.lines[0].set_alpha(0.3)