如何从嵌套列表提取指定元素且避免使用for循环

方案1：纯Python内置函数实现（无需第三方库，适配不等长子列表） #

这种方案利用Python内置的map和zip实现，全程没有Python层面的显式for循环，底层为C实现，执行效率远高于手动遍历，适合绝大多数场景：

# 示例输入
nested_list = [[1,2,3], [3,4,5], [8,6,2,5,6], [7,2,9]]

# 核心处理逻辑
a, b, c = map(list, zip(*map(lambda x: (x[0], x[1], x[2:]), nested_list)))

执行后输出完全符合预期：

• a = [1, 3, 8, 7]
• b = [2, 4, 6, 2]
• c = [[3], [5], [2, 5, 6], [9]]

方案2：Numpy向量化实现（适合超大规模数值型嵌套列表） #

如果你的嵌套列表元素均为数值，且数据量极大，可使用Numpy的向量化操作进一步提升处理效率：

import numpy as np

nested_list = [[1,2,3], [3,4,5], [8,6,2,5,6], [7,2,9]]
nested_arr = np.array(nested_list, dtype=object)

# 提取对应列
a = nested_arr[:, 0].tolist()
b = nested_arr[:, 1].tolist()
c = np.vectorize(lambda x: x[2:])(nested_arr).tolist()

注意：所有子列表需要保证至少包含2个元素，否则会触发索引异常。

内容的提问来源于stack exchange，提问作者Sirui Li。