以下是一个示例代码,用于按列取前n个元素:
def take_n_elements(matrix, n):
result = []
for col in range(len(matrix[0])):
column = [row[col] for row in matrix]
result.append(column[:n])
return result
这个函数接受一个二维矩阵和一个整数n作为参数。它首先创建一个空列表result用于存储结果。然后使用一个循环遍历矩阵的每一列。对于每一列,使用列表解析将该列的元素提取出来,并将提取的列切片为前n个元素。最后将切片后的列添加到结果列表中。最后,返回结果列表。
示例用法:
matrix = [[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]
result = take_n_elements(matrix, 2)
print(result)
输出:
[[1, 4], [2, 5], [3, 6]]
这个示例将矩阵的每一列的前2个元素提取出来,并将结果存储在一个新的二维列表中。
下一篇:按列任意排序矩阵