a-pandas-ex-lookupdict

Creates a lookup dict from a Pandas DataFrame (tuples as keys / np as values)


Keywords
pandas, DataFrame, lookup, dict, python
License
MIT
Install
pip install a-pandas-ex-lookupdict==0.11

Documentation

Creates a lookup dict from a Pandas DataFrame (tuples as keys / np as values)

pip install a-pandas-ex-lookupdict
from a_pandas_ex_lookupdict import pd_add_lookup_dict
import pandas as pd
pd_add_lookup_dict()
df = pd.read_csv(
    "https://raw.githubusercontent.com/pandas-dev/pandas/main/doc/data/titanic.csv"
)
di1 = df[:5].d_get_lookup_dict(as_index=["Name", "PassengerId"])
di2 = df[:5].d_get_lookup_dict(
    as_values=["Cabin", "Fare"], as_index=["Name", "PassengerId"]
)

di1
Out[4]: 
{('Braund, Mr. Owen Harris',
  1): array([[0, 3, 'male', 22.0, 1, 0, 'A/5 21171', 7.25, nan, 'S']],
       dtype=object),
 ('Cumings, Mrs. John Bradley (Florence Briggs Thayer)',
  2): array([[1, 1, 'female', 38.0, 1, 0, 'PC 17599', 71.2833, 'C85', 'C']],
       dtype=object),
 ('Heikkinen, Miss. Laina',
  3): array([[1, 3, 'female', 26.0, 0, 0, 'STON/O2. 3101282', 7.925, nan, 'S']],
       dtype=object),
 ('Futrelle, Mrs. Jacques Heath (Lily May Peel)',
  4): array([[1, 1, 'female', 35.0, 1, 0, '113803', 53.1, 'C123', 'S']],
       dtype=object)}
	   
	   
di2
Out[5]: 
{('Braund, Mr. Owen Harris', 1): array([[nan, 7.25]], dtype=object),
 ('Cumings, Mrs. John Bradley (Florence Briggs Thayer)',
  2): array([['C85', 71.2833]], dtype=object),
 ('Heikkinen, Miss. Laina', 3): array([[nan, 7.925]], dtype=object),
 ('Futrelle, Mrs. Jacques Heath (Lily May Peel)',
  4): array([['C123', 53.1]], dtype=object)}