def caption(text, filename):
display(Markdown('<p class="caption"> <em>{} (<a href="https://github.com/mauforonda/notas/raw/main/{}">imagen en alta resolución</a>) </em> </p>'.format(text, filename)))
def plot_cultivo_municipio(municipio, cultivo):
"""
Crea un gráfico comparativo entre la producción
y superficie cultivada para un cultivo en un municipio
"""
linecolor = '#88888f'
titlecolor = '#1d1f26'
background = '#fcfcfc'
color_produccion = '#f59b42'
color_superficie = '#423e3a'
f, ax = plt.subplots(1,1,figsize=(10,5))
f.set_facecolor(background)
dfi = c[(c.cod_ine == municipio) & (c.cultivo == cultivo)].set_index('year')
dfi.produccion = dfi.produccion / dfi.loc[2013, 'produccion']
dfi.superficie = dfi.superficie / dfi.loc[2013, 'superficie']
ax.plot(dfi.produccion, color=color_produccion, marker='o')
ax.plot(dfi.superficie, color=color_superficie, marker='o')
ax.grid(axis='y')
ax.tick_params(labelcolor=linecolor)
def plot_cultivo_rendimiento(cultivo, ax):
"""
Crea un gráfico comparativo entre el rendimiento de un cultivo entre
todos los municipios donde está presente como múltiplo del rendimiento
en 2013
"""
linecolor = '#88888f'
titlecolor = '#1d1f26'
color_rendimiento = '#9e9b98'
cultivo_df = c[(c.cultivo == cultivo)]
for mun, dfi in cultivo_df.groupby('cod_ine'):
dfi = dfi.set_index('year')
dfi.rendimiento = dfi.rendimiento / dfi.loc[dfi.index.min(), 'rendimiento']
ax.plot(dfi.rendimiento, color=color_rendimiento, marker='o', alpha=.5, markersize=1.2, linewidth=.7)
ax.grid(axis='y')
ax.tick_params(labelcolor=color_rendimiento, labelright=True, labelleft=False, labelsize=10)
ax.annotate(' '.join([word.title() if word not in ['en', 'de', 'con'] else word.lower() for word in cultivo.split()]), xy=(.5,1.02), xycoords='axes fraction', ha='center', va='bottom', fontsize=12, color=color_rendimiento)
last_label = len(ax.get_yticklabels()) - 2
one = [i for i, tick in enumerate(ax.get_yticks()) if tick == 1.]
clear_ticks = [1, last_label] + one
for i, label in enumerate(ax.get_yticklabels()):
if i in clear_ticks:
label.set_alpha(1.)
else:
label.set_alpha(.3)
def get_ranking(c):
"""
Ordena cultivos por la ganancia de rendimiento en el periodo
de estudio, de mayor a menor. Excluye entradas donde el rendimiento
era muy pequeño en 2013, que usualmente apuntan a casos donde no había
producción.
"""
df = pd.DataFrame({i:{'year_min': dfi.year.min(), 'year_max': dfi.year.max(), 'rendimiento_min': dfi[dfi.year == dfi.year.min()].rendimiento.iloc[0], 'rendimiento_max': dfi[dfi.year == dfi.year.max()].rendimiento.iloc[0]} for i, dfi in c.groupby(['cod_ine', 'cultivo'])}).T
df.replace(0, .001, inplace=True)
df['rendimiento_diff'] = df.rendimiento_max / df.rendimiento_min
df = df.reset_index().rename(columns={'level_0': 'cod_ine', 'level_1':'cultivo'})
return df[df.rendimiento_min > .5].groupby('cultivo').rendimiento_diff.mean().sort_values(ascending=False).index.tolist()
def plot_cultivos(c):
"""
Crea un gráfico comparativo para el rendimiento de cada
cultivo como múltiplo del rendimiento en 2013, ordenado
de mayor a menor ganancia de rendimiento.
"""
filename = 'plots/rendimiento_cultivos_2013_2019.png'
ranking = get_ranking(c)
titlecolor = '#1d1f26'
color_rendimiento = '#9e9b98'
f, axs = plt.subplots(6,12,figsize=(50,18), sharex=True)
f.set_facecolor('#fcfcfc')
f.subplots_adjust(hspace=.2, wspace=.2)
axs = axs.flatten()
for cultivo, ax in zip(ranking, axs):
plot_cultivo_rendimiento(cultivo, ax)
axs[0].annotate('Rendimiento de cultivos', xy=(0.0, 1.45), xycoords='axes fraction', fontsize=45, color=titlecolor, ha='left', va='bottom',)
axs[2].annotate('por municipio entre 2013 y 2019 como múltiplo del rendimiento en 2013', xy=(.23, 1.48), xycoords='axes fraction', fontsize=20, color=titlecolor, ha='left', va='bottom',)
plt.savefig(filename, bbox_inches='tight', pad_inches=.4, dpi=200)
plt.show()
caption('Rendimiento por cultivos por municipio entre 2013 y 2019 como múltiplo del rendimiento en 2013', filename)
c = pd.read_csv('data/cultivos.csv')
c = c[c.superficie != 0]
c['rendimiento'] = c.produccion / c.superficie
plot_cultivos(c)
Rendimiento por cultivos por municipio entre 2013 y 2019 como múltiplo del rendimiento en 2013 (imagen en alta resolución)
