main.py

# -*- coding: utf-8 -*-

import pandas as pd

dataset_pd = pd.read_csv(r'./COVID19_line_list_data.csv')

age_of_all_raw = dataset_pd['age'].tolist()
death_or_not_raw = dataset_pd['death'].tolist()

age_of_all = list()

for c in range(0, len(age_of_all_raw)):
  if age_of_all_raw[c] in range(0, 150):
    age_of_all.append(age_of_all_raw[c])

age_of_deaths = list()

for c in range(0, len(age_of_all_raw)):
  if age_of_all_raw[c] in range(0, 150) and death_or_not_raw[c] in (1, '1'):
    age_of_deaths.append(age_of_all_raw[c])

import matplotlib.pyplot as plt

plt.title('BOXPLOT - Comparação de idade entre casos totais \ne casos de óbito em pacientes com COVID-19')
plt.ylabel('Idade dos pacientes')
plt.xlabel('Número de pacientes totais analizados: 1085')
plt.grid('True')
bplot = plt.boxplot([age_of_all, age_of_deaths], labels=['', ''], patch_artist=True)

bplot['boxes'][0].set_facecolor('lightblue')
bplot['boxes'][1].set_facecolor('pink')

import matplotlib.patches as mpatches

lightblue_patch = mpatches.Patch(color='lightblue', label='Casos totais')
pink_patch = mpatches.Patch(color='pink', label='Casos de óbito')
plt.legend(handles=[lightblue_patch, pink_patch], loc='lower center')

#plt.show()
plt.savefig(dpi=400, fname='BOXPLOT-Comparacao-de-idade-entre-casos-totais-e-casos-de-obito-em-pacientes-com-COVID-19.png')