socialfoundations / folktables

Datasets derived from US census data
MIT License
240 stars 19 forks source link

Feature request: Return the BasicProblem as translated dataframes #17

Closed jenno-verdonck closed 2 years ago

jenno-verdonck commented 2 years ago

In the current version of folktables it is possible to return the defined problems as numpy array's. I propose a nice to have feature returning the filtered and processed data as pandas dataframes. This would allow easier formatting and outputting of the problems to csv files for further use in other tools and languages.

A possible extension to this would be to automatically translate the numerical categorical values using the available data dictionaries on the census website. This would make reasoning and analysis of the columns easier.

mrtzh commented 2 years ago

This has now been implemented. See PR #21.