Skip to content

Commit 8d0f030

Browse files
committed
RF: Use list comprehension in reordering
1 parent 78772c4 commit 8d0f030

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

nibabel/nifti1.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2014,13 +2014,9 @@ def as_reoriented(self, ornt):
20142014
return img
20152015

20162016
# Also apply the transform to the dim_info fields
2017-
new_dim = list(img.header.get_dim_info())
2018-
for idx, value in enumerate(new_dim):
2019-
# For each value, leave as None if it was that way,
2020-
# otherwise check where we have mapped it to
2021-
if value is None:
2022-
continue
2023-
new_dim[idx] = ornt[value, 0]
2017+
new_dim = [
2018+
None if orig_dim is None else int(ornt[orig_dim, 0])
2019+
for orig_dim in img.header.get_dim_info()]
20242020

20252021
img.header.set_dim_info(*new_dim)
20262022

0 commit comments

Comments
 (0)