def confirmados_diarios_por_estado(datos, entidades):
"""
Calcula el número total de casos confirmados por fecha y por estado.
Input:
- datos: datos abiertos de COVID-19 en México disponibles en [1].
Output:
- serie: Serie de tiempo de nuevos casos confirmados por dia para cada
entidad federativa en México.
[1]: https://www.gob.mx/salud/documentos/datos-abiertos-152127
"""
series = (datos[datos['CLASIFICACION_FINAL'].isin([1, 2, 3])]
.groupby(['ENTIDAD_UM', 'FECHA_INGRESO'])
.count()['ORIGEN'])
return get_formato_series(series, entidades)
def sospechosos_diarios_por_estado(datos, entidades):
"""
Calcula el número de sospechosos por fecha y por estado.
Input:
- datos: datos abiertos de COVID-19 en México disponibles en [1].
Output:
- series: Serie de tiempo de nuevas pruebas pendientes por dia para cada
entidad federativa en México.
[1]: https://www.gob.mx/salud/documentos/datos-abiertos-152127
"""
idx = datos['CLASIFICACION_FINAL'].isin([3, 4, 5])
series = (datos.loc[idx].groupby(['ENTIDAD_UM',
'FECHA_INGRESO']).count()['ORIGEN'])
return get_formato_series(series, entidades)
def defunciones_diarias_por_estado(datos, entidades):
"""
Calcula el número de defunciones por fecha y por estado.
Input:
- datos: datos abiertos de COVID-19 en México disponibles en [1].
Output:
- series: Serie de tiempo de nuevas muertes por dia para cada entidad
federativa en México.
[1]: https://www.gob.mx/salud/documentos/datos-abiertos-152127
"""
idx = (datos['CLASIFICACION_FINAL'].isin([1, 2, 3])) & (datos['FECHA_DEF'] != '9999-99-99')
series = (datos[idx]
.groupby(['ENTIDAD_UM', 'FECHA_DEF'])
.count()['ORIGEN'])
return get_formato_series(series, entidades)
def uci_diarios_por_estado(datos, entidades):
"""
Calcula el número de pacientes ingresados a una UCI por fecha y por estado.
Input:
- datos: datos abiertos de COVID-19 en México disponibles en [1].
Output:
- series: Serie de tiempo de nuevos pacientes en UCI por dia para cada
entidad federativa en México.
[1]: https://www.gob.mx/salud/documentos/datos-abiertos-152127
"""
idx = (datos['CLASIFICACION_FINAL'].isin([1, 2, 3]) & (datos['UCI'] == 1))
series = (datos[idx]
.groupby(['ENTIDAD_UM', 'FECHA_INGRESO'])
.count()['ORIGEN'])
return get_formato_series(series, entidades)
def pruebas_totales_diarias_por_estado(datos, entidades):
"""
Calcula el número total de pruebas realizadas por fecha y por estado.
Input:
- datos: datos abiertos de COVID-19 en México disponibles en [1].
Output:
- series: Serie de tiempo de nuevas pruebas totales por dia para cada
entidad federativa en México.
[1]: https://www.gob.mx/salud/documentos/datos-abiertos-152127
"""
idx = (datos['TOMA_MUESTRA_LAB'] == 1) | (datos['TOMA_MUESTRA_ANTIGENO']
== 1)
series = (datos.loc[idx].groupby(['ENTIDAD_UM',
'FECHA_INGRESO']).count()['ORIGEN'])
return get_formato_series(series, entidades)