python定义函数

Question

完成transform_2d_list()函数，该函数接受一个二维列表作为参数a_2d_list。参数列表有m个项，其中每个项是包含n个项的列表。该函数返回一个新的二维列表，其中第一项是包含原始列表中列表项中的所有第一项的列表，第二项是包含原始列表中列表项中的所有第二项的列表，依此类推。换句话说，返回的二维列表将有n个条目，其中每个条目都是包含m个条目的列表。你可以假设m和n都至少是1。
下面显示了被调用函数的一些示例：
输入：a_2d_list = [[1]]
print(transform_2d_list(a_2d_list))
输出： [[1]]
输入：a_2d_list = [['a'], ['b']]
print(transform_2d_list(a_2d_list))
输出： [['a', 'b']]
输入： a_2d_list = [[1, 2], [3, 4]]
print(transform_2d_list(a_2d_list))
输出： [[1, 3], [2, 4]]

def transform_2d_list(a_2d_list)：

Answer 1

def transform_2d_list(a_2d_list):
    return [[k[i] for k in a_2d_list] for i in range(len(a_2d_list[0]))]

print(transform_2d_list( [[1, 2], [3, 4], [5, 6]]))
--result
[[1, 3, 5], [2, 4, 6]]