A (key-value) data-object-layer to get (pandas) tables from a variety of sources with ease
To install: pip install tabled
Tabled provides seamless integration with SQLite databases through DfFiles:
fromtabledimportDfFiles# Automatic SQLite detection - just pass the database file pathdf_files=DfFiles('my_database.db')
# Access tables as DataFramescustomers=df_files['customers.parquet'] # Full filenameorders=df_files['orders'] # Clean table name (both work)# List available tablesprint(list(df_files.keys())) # ['customers.parquet', 'orders.parquet', ...]# Or use the explicit methoddf_files=DfFiles.from_sqlite_file('my_database.db')Under the hood, SQLite tables are exported to temporary Parquet files for efficient access, with automatic cleanup when the program exits.
For more control over SQLite data extraction, use the sqlite_tools module:
fromtabled.sqlite_toolsimportexport_sqlite_to_dataframes, export_sqlite_to_parquet# Export to DataFramestables=export_sqlite_to_dataframes('database.db')
customers_df=tables['customers']
# Export to Parquet filesexport_sqlite_to_parquet('database.db', 'output_directory/')The dataframe_info function provides flexible analysis of pandas DataFrames:
fromtabled.diagnoseimportdataframe_info, register_info_funcimportpandasaspd# Analyze a DataFramedf=pd.DataFrame({'a': [1, 2, 3], 'b': ['x', 'y', 'z']})
info=dataframe_info(df)
print(info['shape']) # (3, 2)print(info['columns']) # ['a', 'b']# Extend with custom analysis functionsdefmemory_usage(df):
returndf.memory_usage(deep=True).sum()
register_info_func('memory', memory_usage)
info=dataframe_info(df)
print(info['memory']) # Memory usage in bytesThe analysis is completely customizable - you can register new analysis functions or provide custom info function dictionaries to focus on specific aspects of your data.
This section demonstrates how to use DfFiles to store and retrieve pandas DataFrames using various file formats.
First, let's import required packages and define our test data:
importosimportshutilimporttempfileimportpandasaspdfromtabledimportDfFiles# Test data dictionarymisc_small_dicts= {
"fantasy_tavern_menu": {
"item": ["Dragon Ale", "Elf Bread", "Goblin Stew"],
"price": [7.5, 3.0, 5.5],
"is_alcoholic": [True, False, False],
"servings_left": [12, 25, 8],
},
"alien_abduction_log": {
"abductee_name": ["Bob", "Alice", "Zork"],
"location": ["Kansas City", "Roswell", "Jupiter"],
"duration_minutes": [15, 120, 30],
"was_returned": [True, False, True],
}
}We'll create a temporary directory for our files:
defcreate_test_directory():
# Create a directory for the test filesrootdir=os.path.join(tempfile.gettempdir(), 'tabled_df_files_test')
ifos.path.exists(rootdir):
shutil.rmtree(rootdir)
os.makedirs(rootdir)
print(f"Created directory at: {rootdir}")
returnrootdirrootdir=create_test_directory()
print(f"Created directory at: {rootdir}")Created directory at: /var/folders/mc/c070wfh51kxd9lft8dl74q1r0000gn/T/tabled_df_files_test
Created directory at: /var/folders/mc/c070wfh51kxd9lft8dl74q1r0000gn/T/tabled_df_files_test
Create a new DfFiles instance pointing to our directory:
df_files=DfFiles(rootdir)Let's verify it starts empty:
list(df_files)[]
Let's create DataFrames from our test data:
fantasy_tavern_menu_df=pd.DataFrame(misc_small_dicts['fantasy_tavern_menu'])
alien_abduction_log_df=pd.DataFrame(misc_small_dicts['alien_abduction_log'])
print("Fantasy Tavern Menu:")
display(fantasy_tavern_menu_df)
print("\nAlien Abduction Log:")
display(alien_abduction_log_df)Fantasy Tavern Menu:
| item | price | is_alcoholic | servings_left | |
|---|---|---|---|---|
| 0 | Dragon Ale | 7.5 | True | 12 |
| 1 | Elf Bread | 3.0 | False | 25 |
| 2 | Goblin Stew | 5.5 | False | 8 |
Alien Abduction Log:
| abductee_name | location | duration_minutes | was_returned | |
|---|---|---|---|---|
| 0 | Bob | Kansas City | 15 | True |
| 1 | Alice | Roswell | 120 | False |
| 2 | Zork | Jupiter | 30 | True |
Now let's save these DataFrames using different formats:
df_files['fantasy_tavern_menu.csv'] =fantasy_tavern_menu_dfdf_files['alien_abduction_log.json'] =alien_abduction_log_dfLet's verify we can read the data back correctly:
saved_df=df_files['fantasy_tavern_menu.csv']
saved_df| item | price | is_alcoholic | servings_left | |
|---|---|---|---|---|
| 0 | Dragon Ale | 7.5 | True | 12 |
| 1 | Elf Bread | 3.0 | False | 25 |
| 2 | Goblin Stew | 5.5 | False | 8 |
DfFiles implements the MutableMapping interface, making it behave like a dictionary.
Let's see how many files we have:
len(df_files)2
List all available files:
list(df_files)['fantasy_tavern_menu.csv', 'alien_abduction_log.json']
Check if a file exists:
'fantasy_tavern_menu.csv'indf_filesTrue
Let's see what file formats DfFiles supports out of the box.
(Note that some of these will require installing extra packages, which you'll realize if you get an ImportError)
print("Encoder supported extensions:")
list_of_encoder_supported_extensions=list(df_files.extension_encoder_mapping)
print(*list_of_encoder_supported_extensions, sep=', ')Encoder supported extensions:
csv, txt, tsv, json, html, p, pickle, pkl, npy, parquet, zip, feather, h5, hdf5, stata, dta, sql, sqlite, gbq, xls, xlsx, xml, orc
print("Decoder supported extensions:")
list_of_decoder_supported_extensions=list(df_files.extension_decoder_mapping)
print(*list_of_decoder_supported_extensions, sep=', ')Decoder supported extensions:
csv, txt, tsv, parquet, json, html, p, pickle, pkl, xml, sql, sqlite, feather, stata, dta, sas, h5, hdf5, xls, xlsx, orc, sav
Let's try saving and loading our test DataFrame in different formats:
extensions_supported_by_encoder_and_decoder= (
set(list_of_encoder_supported_extensions) &set(list_of_decoder_supported_extensions)
)
sorted(extensions_supported_by_encoder_and_decoder)['csv',
'dta',
'feather',
'h5',
'hdf5',
'html',
'json',
'orc',
'p',
'parquet',
'pickle',
'pkl',
'sql',
'sqlite',
'stata',
'tsv',
'txt',
'xls',
'xlsx',
'xml']
deftest_extension(ext):
filename=f'test_file.{ext}'try:
df_files[filename] =fantasy_tavern_menu_dfdf_loaded=df_files[filename]
# test the decoded df is the same as the one that was saved (round-trip test)# Note that we drop the index, since the index is not saved in the file by default for all codecspd.testing.assert_frame_equal(
fantasy_tavern_menu_df.reset_index(drop=True),
df_loaded.reset_index(drop=True),
)
returnTrueexceptExceptionase:
returnFalsetest_extensions= [
'csv',
'feather',
'json',
'orc',
'parquet',
'pkl',
'tsv', # 'dta', # TODO: fix# 'h5', # TODO: fix# 'html', # TODO: fix# 'sql', # TODO: fix# 'xml', # TODO: fix
]
forextintest_extensions:
print("Testing extension:", ext)
success=test_extension(ext)
ifsuccess:
print(f"\tExtension {ext}: ✓")
else:
print('\033[91m'+f"\tFix extension {ext}: ✗"+'\033[0m')
# marker = '✓' if success else '\033[91m✗\033[0m'# print(f"\tExtension {ext}: {marker}")Testing extension: csv
Extension csv: ✓
Testing extension: feather
Extension feather: ✓
Testing extension: json
Extension json: ✓
Testing extension: orc
Extension orc: ✓
Testing extension: parquet
Extension parquet: ✓
Testing extension: pkl
Extension pkl: ✓
Testing extension: tsv
Extension tsv: ✓
Testing extension: dta
�[91m Fix extension dta: ✗�[0m
Testing extension: h5
�[91m Fix extension h5: ✗�[0m
Testing extension: html
�[91m Fix extension html: ✗�[0m
Testing extension: sql
�[91m Fix extension sql: ✗�[0m
Testing extension: xml
�[91m Fix extension xml: ✗�[0m