python - 在嵌套列表中搜索和替换值的方法

Question

我想搜索和替换列表列表中的值。我拼凑了以下问题的答案:

1. Flatten a nested list
2. Search for and replace a value
3. Regroup the flat list into a list of lists

我当前的代码有效，但我觉得它比需要的更复杂。有更优雅的方法吗？

# Create test data- a list of lists which each contain 2 items
numbers = list(range(10))
list_of_lists = [numbers[i:i+2] for i in range(0, len(numbers), 2)]

# Flatten the list of lists
flat_list = [item for sublist in list_of_lists for item in sublist]
# Search for and replace values
modified_list = [-1 if e > 5 else e for e in flat_list]
# Regroup into a list of lists
regrouped_list_of_lists = [modified_list[i:i+2] for i in range(0, len(modified_list), 2)]

Answer 1

在嵌套列表理解中进行子列表的替换，而无需展平和重新组合:

numbers = list(range(10))
list_of_lists = [numbers[i:i+2] for i in range(0, len(numbers), 2)]
# here
list_of_lists = [[-1 if e > 5 else e for e in sublist] for sublist in list_of_lists]