如何用Python普通列表实现Numpy[:,0]式的二维索引？

首先明确：Python原生列表不支持Numpy那种[:,0]式的多维切片语法，你的语法本身没问题，但pylist[:][0]的行为和你预期的不一样。

为什么pylist[:][0]不符合预期？ #

pylist[:]是对整个列表做一次浅拷贝，返回的是和原列表内容完全一致的新列表；之后再加[0]，只是取这个新列表的第一个元素，自然还是[1,2,3]，根本没遍历子列表去取每个子列表的第一个元素。

替代方案 #

针对普通嵌套列表（包括非矩形的动态列表），有几种简单实用的方法可以获取每个子列表的第一个元素：

1. 列表推导式（最推荐）

这是Python里最直观、可读性最高的方式，还能灵活处理可能存在的空子列表：

pylist = [[1,2,3],[4,5,6],[7,8,9]]
# 获取每个子列表第一个元素
first_elements = [sub[0] for sub in pylist]
print(first_elements)  # 输出: [1,4,7]

# 处理包含空列表的场景
pylist_with_empty = [[1,2], [], [3,4,5]]
first_elements_safe = [sub[0] for sub in pylist_with_empty if sub]
print(first_elements_safe)  # 输出: [1,3]

2. 使用map函数

如果偏好函数式风格，可以用map配合匿名函数实现：

pylist = [[1,2,3],[4,5,6],[7,8,9]]
first_elements = list(map(lambda x: x[0], pylist))
print(first_elements)  # 输出: [1,4,7]

# 处理空列表的安全版本
first_elements_safe = list(map(lambda x: x[0] if x else None, pylist_with_empty))
print(first_elements_safe)  # 输出: [1, None, 3]

这两种方法都能完美适配非矩形的动态列表，比如子列表长度各不相同的情况，完全可以替代Numpy的多维索引功能。

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