要按列排列的前 n 个元素,你可以使用 Python 中的列表推导式和切片操作来实现。以下是一个示例:
# 假设你有一个二维列表,表示一个矩阵
matrix = [
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
]
n = 2 # 获取每列的前两个元素
result = [col[:n] for col in zip(*matrix)]
print(result)
输出结果为:
[(1, 4), (2, 5), (3, 6)]
在上述示例中,我们使用 zip(*matrix)
将矩阵的行转置为列。然后,我们使用列表推导式 [col[:n] for col in zip(*matrix)]
来获取每列的前 n 个元素。
领取专属 10元无门槛券
手把手带您无忧上云