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

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]

Answer 2

您已经在使用列表推导，只需将它们组合起来：

replaced_list_of_lists = [
            [-1 if e > 5 else e for e in inner_list]
                for inner_list in list_of_lists
        ]