Cómo convertir datos Csv en un diccionario
Utilicé Pandas para leer el archivo CSV y pude crear un diccionario a partir de él, pero necesito crear el diccionario en un formato específico.
flightinfo = pd.read_csv('flightdata.csv', quotechar='"', names=['startingairport', 'destinationairport', 'airline', 'time', 'startstate', 'end_state'], header=None)
flightinfodict = flightinfo.to_dict(orient='records')
Esto da como resultado:
[{'startingairport': 'Hilo International Airport', 'destinationairport': 'Boise Airport ', 'airline': 'delta', 'averagetimetaken': 300, 'startstate': 'HAWAII', 'endstate': 'IDAHO'}
Pero el resultado que necesito es:
{'Hilo International Airport': {'destinationairport': 'Boise Airport ', 'airline': 'delta', 'averagetime_taken': 300}}
¿Cómo puedo modificar el código para producir este resultado?
Gracias.
davy.ai
Puedes usar una comprensión de diccionario para crear la salida deseada:
flightinfodict = {fila['starting_airport']: {'destination_airport': fila['destination_airport'], 'airline': fila['airline'], 'average_time_taken': fila['time'], 'start_state': fila['start_state'], 'end_state': fila['end_state']} for fila in flightinfo.to_dict(orient='records')}