如何在Python中针对不等长子列表编写列表推导式？

嘿，我来帮你搞定这个问题！你遇到的IndexError几乎肯定是因为在列表推导式里，你尝试访问了某些子列表根本不存在的索引位置——毕竟你的子列表长度都不一样，总有个子列表会比你要访问的索引短嘛。下面给你几种常见场景的解决办法：

场景1：提取特定索引的元素，但部分子列表长度不足 #

比如你想提取每个子列表的第1个元素（索引为1），但有的子列表只有1个元素（索引只能到0），这时候直接写sublist[1]就会报错。

解决办法： #

• 过滤掉长度不够的子列表：只处理那些长度满足要求的子列表，避免访问无效索引：

  sublists = [[1, 2], [3], [4, 5, 6]]
  # 只提取长度>=2的子列表的索引1元素
  new_list = [sublist[1] for sublist in sublists if len(sublist) > 1]
  # 输出结果：[2, 5]
  

• 给短子列表设默认值：如果不想过滤掉，而是给长度不足的子列表补一个默认值（比如None）：

  sublists = [[1, 2], [3], [4, 5, 6]]
  new_list = [sublist[1] if len(sublist) > 1 else None for sublist in sublists]
  # 输出结果：[2, None, 5]
  

场景2：按位置对齐合并子列表（类似转置） #

如果你想把所有子列表的第0个元素放一组，第1个放一组……但因为子列表长度不一，后面的位置有些子列表没有元素，直接用zip会截断到最短的子列表，而用索引循环又会报错。

解决办法：用itertools.zip_longest处理不等长序列 #

zip_longest会自动用指定的填充值补上缺失的元素，再结合列表推导式生成结果：

from itertools import zip_longest

sublists = [[1, 2], [3], [4, 5, 6]]
# 按位置分组，缺失的元素用None填充
new_list = [list(group) for group in zip_longest(*sublists, fillvalue=None)]
# 输出结果：[[1, 3, 4], [2, None, 5], [None, None, 6]]

核心思路总结 #

不管哪种场景，本质都是要避免访问子列表不存在的索引：

• 要么提前判断子列表长度，只处理符合条件的；
• 要么用工具函数（比如zip_longest）帮你自动处理不等长的情况，不用自己手动索引。

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