More pythonic version of change
This commit is contained in:
1 parent
089594f72d
commit
da4404e875
1 file changed
+1
-1
@@ -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):
|
||||
|
||||
Reference in new issue
Block a user