[ Foro de Python ]

PANDAS

import pandas as pd
import os, sys

wrkspc = r'C:\Users\eduardo.varela\Desktop\Ejercicios Python'
fileInput = input("Ingrese nombre del archivo")
fileOutput = fileInput + "V2"
dias = {"LUNES": "01 LUNES", "MARTES": "02 MARTES", "MIERCOLES": "03 MIERCOLES",

       "JUEVES": "04 JUEVES", "VIERNES": "05 VIERNES", "SABADO": "06 SABADO",
       "DOMINGO": "07 DOMINGO"}

colDSemana = "DIA_SEMAN"
colComuna = "COMUNA"
colMes = "MES_DEL_HE"
colFecha = "FECHA_DEL"
colSemana = "Semana"
distritos = {"Distrito 1": [1, 2, 3, 4, 9], "Distrito 2": [5, 6, 7, 8, 11, 12], "Distrito 3":[10, 17, 18, 19, 20],
            "Distrito 4": [13, 14, 15, 16, 21], "Distritos 5": [22]}

df = pd.read_excel(os.path.join(wrkspc, fileInput+".xlsx"), converters={colFecha:pd.to_datetime})
df[colComuna] = df[colComuna].replace(" ", "0").fillna(0).apply(int)
for comunas in distritos:
   df.loc[df[colComuna].isin(distritos[comunas]), "Distrito"] = comunas

df.Distrito.fillna("Rural", inplace=True)

for dia in dias:
   df.loc[df[colDSemana].str.upper() == dia, "diasemana"] = dias[dia]


df[colMes] = df[colFecha].dt.month
df[colSemana] = df[colFecha].dt.week
print(df.columns)
with pd.ExcelWriter(os.path.join(wrkspc, fileOutput+".xlsx")) as writer:
   df.to_excel(writer, "Delitos")
   del df