Enable dictionary format on `dask.dataframe.DataFrame.to_bag` output

The current dask.dataframe.DataFrame.to_bag behavior is as follow:

import pandas as pd
import dask.dataframe as dd
import dask.bag as db

my_dict = {
    'Name' : ['John', 'Lara', 'James', 'Pablo'],
    'Age' : [23, 21, 22, 21],
    'Fruit' : ['orange', 'apple', 'apricot', 'cherry'],
}

df = pd.DataFrame(my_dict)
ddf = dd.from_pandas(df, npartitions=3)

ddf_bag = ddf.to_bag()
ddf_bag.compute()

[('John', 23, 'orange'),
 ('Lara', 21, 'apple'),
 ('James', 22, 'apricot'),
 ('Pablo', 21, 'cherry')]

This loses the information of the column names, I would like to see something like this as an output

[{'name': 'John', 'Age': 23, 'Fruit':'orange'},
 {'name':'Lara', 'Age': 21, 'Fruit':'apple'},
 {'name':'James', 'Age': 22, 'Fruit':'apricot'},
 {'name':'Pablo', 'Age':21, 'Fruit':'cherry'}]

[jsignell]

Closing this - thanks @rajagurunath

[mrocklin]

Welcome! I encouraged you to submit a pull request, even if it is not done yet. Pull requests make it easier for maintainers to view and to engage in conversation.

On Sat, Jul 24, 2021, 1:13 PM gurunath @.***> wrote:

Hi @jrbourbeau https://github.com/jrbourbeau @mrocklin https://github.com/mrocklin @ncclementi https://github.com/ncclementi

Thanks for the detailed discussion, based on the comments above, implemented the initial draft version here: @.*** https://github.com/rajagurunath/dask/commit/ce6812f69c3f0384e79b2927addbd0025d6bf520

I am kind of new to the dask community, Can someone have a look into the above commit and help me with further suggestions/Best practices, etc. Very much happy to take this as my good first issue ❤️

Thanks in Advance

— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub https://github.com/dask/dask/issues/7920#issuecomment-886090922, or unsubscribe https://github.com/notifications/unsubscribe-auth/AACKZTGOHYG2YIG2L2PSACLTZL7EZANCNFSM5AWZ7B6Q .