python怎么提取某一列每行的前几个元素

假设你有一个二维列表`matrix`，可以使用列表推导式提取每行前几个元素组成一个新的列表，例如提取第一列前两个元素，可以这样写：

col_1 = [row[0:2] for row in matrix]

其中`row[0:2]`表示提取`row`这一行的第0个和第1个元素。如果想提取其他列前几个元素，只需要调整`row`中的下标即可。