Source code for flipper.census
''' A module for loading flipper databases. '''
import pandas as pd
import os
DATABASE_DIRECTORY = os.path.join(os.path.dirname(__file__), 'censuses')
DATABASES = set(os.path.splitext(os.path.basename(path))[0] for path in os.listdir(DATABASE_DIRECTORY) if os.path.splitext(path)[1] == '.csv')
[docs]def census(census_name):
''' Return the requsted database. '''
if census_name in DATABASES:
census_name = os.path.join(DATABASE_DIRECTORY, census_name + '.csv')
return pd.read_csv(census_name)