Skip to content

⚡ Bolt: [performance improvement] Replace iterrows with to_dict('records') #67

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
alinelena wants to merge 1 commit into
base: main
Choose a base branch
from
Open

⚡ Bolt: [performance improvement] Replace iterrows with to_dict('records') #67

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions .jules/bolt.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,3 +8,6 @@
## 2024-03-29 - ASE Custom JSON encoding vs standard JSON
**Learning:** ASE's custom JSON encoder (`ase.io.jsonio.encode`) will generate dicts with special keys like `__ndarray__` or `__complex__` (e.g. `{"__ndarray__": [[5], "int64", ...]}`). When optimizing JSON deserialization using faster alternatives like `orjson`, it's critical to realize that a normal `json.loads` or `orjson.loads` will deserialize this into a Python dictionary, while ASE's custom `decode` will properly reconstruct the underlying numpy array. Bypassing ASE's decoder without checking for these keys leads to downstream type errors (e.g. `KeyError: '__ndarray__'`).
**Action:** When replacing or wrapping ASE's jsonio with `orjson`, always fall back to ASE's `decode` if the payload string contains `__ndarray__` or `__complex__` markers, to ensure custom objects are correctly reconstructed.
## 2024-06-20 - Replace df.iterrows() with df.to_dict('records')
**Learning:** In data processing scripts like `verify_processed_omol25.py`, iterating over large Pandas DataFrames using `df.iterrows()` is a major anti-pattern that creates significant bottlenecks due to the overhead of boxing rows into Pandas Series objects.
**Action:** When iterating over DataFrames row-by-row, always use `df.to_dict('records')` first to convert the DataFrame to a list of native Python dictionaries, which provides a massive performance speedup (often 10x-100x depending on row count).
7 changes: 4 additions & 3 deletions src/lavello_mlips/verify_processed_omol25.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -77,8 +77,9 @@ def main() -> None:
)

logger.info(f"Loaded {len(df)} records from Parquet.")
parquet_by_sha = {row["geom_sha1"]: row for _, row in df.iterrows()}
parquet_by_argone_rel = {row["argonne_rel"]: row for _, row in df.iterrows()}
records = df.to_dict("records")
parquet_by_sha = {row["geom_sha1"]: row for row in records}
parquet_by_argone_rel = {row["argonne_rel"]: row for row in records}
logger.info(f"Loading ExtXYZ file from {args.extxyz} (this may take a moment)...")
all_atoms = read(str(args.extxyz), index=":")
if not isinstance(all_atoms, list):
Expand Down Expand Up @@ -108,7 +109,7 @@ def get_dump_entry(at):
info = dict(at.info)
rel = info.get("argonne_rel")
pq_row = parquet_by_argone_rel.get(rel)
pq_data = pq_row.to_dict() if pq_row is not None else None
pq_data = dict(pq_row) if pq_row is not None else None
return {"xyz": info, "parquet": pq_data}

duplicates = {
Expand Down
Loading