Python中按数值条件拆分列表子块，是否有单行实现语句或函数？

当然有办法用单行语句实现这个需求！我们可以结合列表推导式和标准库工具来完成，下面给你两种可行的方案：

方案1：使用itertools工具函数（更高效） #

借助itertools.takewhile和itertools.dropwhile可以快速拆分出符合条件的子块，代码如下：

from itertools import takewhile, dropwhile

original_list = [[2, 3, 4], [3,4,5,6,7]]
result = [sub_block for item in original_list for sub_block in ([item] if all(x <= 4 for x in item) else [list(takewhile(lambda x: x <= 4, item)), list(dropwhile(lambda x: x <= 4, item))])]

逻辑说明： #

• 遍历原列表中的每个子列表
• 如果子列表所有元素都≤4，直接保留该子列表
• 否则，用takewhile提取出第一个大于4的元素之前的所有元素，用dropwhile去掉前面≤4的元素得到剩余部分，将这两个新子块加入结果

方案2：纯列表推导式（无需额外导入） #

如果不想导入itertools，也可以通过查找第一个大于4的元素索引来拆分，代码如下：

original_list = [[2, 3, 4], [3,4,5,6,7]]
result = []
[result.extend([item] if all(x <= 4 for x in item) else [item[:next(i for i, x in enumerate(item) if x > 4)], item[next(i for i, x in enumerate(item) if x > 4):]]) for item in original_list]

注意点： #

这个方案会两次查找第一个大于4的元素索引，效率略低于第一种，但胜在无需导入任何模块。另外，这两种方案都默认你的子列表中≤4的元素是连续出现在开头的（和你的示例一致），如果子列表中存在穿插的大于4的元素，可能需要调整逻辑来匹配你的实际需求。

内容的提问来源于stack exchange，提问作者Naaga A