Eu tenho um quadro de dados do pandas com várias colunas e gostaria de construir um dict a partir de duas colunas: uma como as chaves do dict e a outra como os valores do dict. Como eu posso fazer isso?
Quadro de dados:
area count
co tp
DE Lake 10 7
Forest 20 5
FR Lake 30 2
Forest 40 3
Preciso definir a área como chave, contar como valor em dict. Agradeço antecipadamente.
python
pandas
dataframe
dictionary
data-conversion
perigeu
fonte
fonte
TypeError: zip argument #2 must support iteration
area_dict = dict(zip(lakes['area'], lakes['count']))
area_dict = dict(zip(lakes.area, (lakes.count, lakes.other_column)))
. Como você faria isso acontecer?Com os pandas, pode ser feito como:
Se lakes for seu DataFrame:
area_dict = lakes.to_dict('records')
fonte
orient
.Você também pode fazer isso se quiser brincar com os pandas. No entanto, gosto do jeito do punchagan.
# replicating your dataframe lake = pd.DataFrame({'co tp': ['DE Lake', 'Forest', 'FR Lake', 'Forest'], 'area': [10, 20, 30, 40], 'count': [7, 5, 2, 3]}) lake.set_index('co tp', inplace=True) # to get key value using pandas area_dict = lake.set_index('area').T.to_dict('records')[0] print(area_dict) output: {10: 7, 20: 5, 30: 2, 40: 3}
fonte