"Si un trabajador quiere hacer bien su trabajo, primero debe afilar sus herramientas." - Confucio, "Las Analectas de Confucio. Lu Linggong"
Página delantera > Programación > ¿Cómo agrupar datos por columna y convertirlos en un diccionario de listas en Python?

¿Cómo agrupar datos por columna y convertirlos en un diccionario de listas en Python?

Publicado el 2024-11-08
Navegar:131

How to Group Data by Column and Convert it to a Dictionary of Lists in Python?

Resultados de GroupBy en diccionario de listas

Tiene una hoja de cálculo de Excel con tres columnas, Columna1, Columna2 y Columna3. Desea extraer los datos, agruparlos por Columna 1 y crear un diccionario de listas similar a este:

{0: [1],
1: [2, 3, 5],
2: [1, 2],
3: [4, 5],
4: [1],
5: [1, 2, 3]}

Para lograr este resultado, puede utilizar el siguiente código:

excel = pandas.read_excel(r"e:\test_data.xlsx", sheetname='mySheet', parse_cols="A,C")
grouped_data = excel.groupby("Column1")["Column3"].apply(list).to_dict()

La función groupby agrupa los datos por Columna1 y devuelve un objeto groupby. Luego, el método de aplicación se utiliza para aplicar la función de lista a los datos de la Columna 3, creando una lista de valores para cada grupo. Finalmente, el método to_dict se utiliza para convertir los datos agrupados en un diccionario de listas.

El diccionario resultante tendrá el formato que usted especificó, siendo las claves los valores de la Columna 1 y los valores las listas de la valores correspondientes en la columna 3.

Último tutorial Más>

Descargo de responsabilidad: Todos los recursos proporcionados provienen en parte de Internet. Si existe alguna infracción de sus derechos de autor u otros derechos e intereses, explique los motivos detallados y proporcione pruebas de los derechos de autor o derechos e intereses y luego envíelos al correo electrónico: [email protected]. Lo manejaremos por usted lo antes posible.

Copyright© 2022 湘ICP备2022001581号-3