Skip to content

Commit e2c7038

Browse files
Fix pd.DataFrame indexing in convert_tables_to_dicts()
Resolves: #8201 Signed-off-by: Bartosz Grabowski <[email protected]>
1 parent 44add8d commit e2c7038

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

monai/data/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1484,7 +1484,7 @@ def convert_tables_to_dicts(
14841484
rows.append(i)
14851485

14861486
# convert to a list of dictionaries corresponding to every row
1487-
data_ = df.loc[rows] if col_names is None else df.loc[rows, col_names]
1487+
data_ = df.iloc[rows] if col_names is None else df.iloc[rows][col_names]
14881488
if isinstance(col_types, dict):
14891489
# fill default values for NaN
14901490
defaults = {k: v["default"] for k, v in col_types.items() if v is not None and v.get("default") is not None}

0 commit comments

Comments
 (0)