Skip to content

Instantly share code, notes, and snippets.

@kyo-takano

kyo-takano/pandas-concise-query.py

Created March 16, 2024 03:16
Show Gist options
  • Save kyo-takano/4d1ba88cdc01f19e38aa3799ec0a93f3 to your computer and use it in GitHub Desktop.
Save kyo-takano/4d1ba88cdc01f19e38aa3799ec0a93f3 to your computer and use it in GitHub Desktop.
Download ZIP
Creating a concise query for Pandas DataFrame
Raw
pandas-concise-query.py
init_std, lr = 0.02, 2e-5 # example
""" Redundant """
indices = df["init_std"] == init_std) & (df["lr"] == lr)
print(df[(indices])
""" Concise """
query = f"{init_std=} & {lr=}".replace("=", "==")
print(df.query(query)) # "init_std==0.02 & lr==2e-05"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment