More pythonic version of change

This commit is contained in:
Raif Olson committed 2024-03-28 08:38:50 -04:00
1 parent 089594f72d
commit da4404e875
1 file changed
+1 -1
+1 -1
View File
@@ -849,7 +849,7 @@ def get_data_item(
subset_data[key] = [value[i] for i in index]
elif isinstance(index, np.ndarray):
if index.dtype == bool:
subset_data[key] = [value[i] for i in range(len(index)) if index[i]]
subset_data[key] = [value[i] for i, index_value in enumerate(index) if index_value]
else:
subset_data[key] = [value[i] for i in index]
elif isinstance(index, int):