Skip to content

lpenguin/pandas-query

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pandas-query module

import pandas as pd
from pandas_query import _

really_long_name_dataframe = pd.DataFrame({'ints': range(10)})

Assigning new columns

# Instead of 
really_long_name_dataframe['mul10'] = really_long_name_dataframe['ints'] * 10
really_long_name_dataframe['squares'] = really_long_name_dataframe['ints'] ** 2

# Write shorter
really_long_name_dataframe['mul10'] = _['ints'] * 10
really_long_name_dataframe['squares'] = _['ints'] ** 2

Selecting data

# Instead of
subset = really_long_name_dataframe[
    really_long_name_dataframe[['ints'].between(3, 6) 
    & (really_long_name_dataframe[['mul10'] != 40)
]

# Write shorter
subset = really_long_name_dataframe[
    _['ints'].between(3, 6) 
    & (_['mul10'] != 40)
]

Evaluating expressions

# Instead of
cubes = (
    really_long_name_dataframe['ints'] * really_long_name_dataframe['squares'] 
)

# Write shorter
cubes = (
    really_long_name_dataframe(_['ints'] * _['squares']) 
)

About

Shorter pandas expressions and indexing. Proof of concept

Topics

Resources

Stars

Watchers

Forks

Packages

No packages published

Contributors 2

  •  
  •