【leetcode】867 转置矩阵

• 题目

给定一个矩阵 A， 返回 A 的转置矩阵。
矩阵的转置是指将矩阵的主对角线翻转，交换矩阵的行索引与列索引。

• 我的答案

class Solution:
    def transpose(self, A: List[List[int]]) -> List[List[int]]:
        trans = []
        for column in range(len(A[0])):
            temp_list = []
            for row in range(len(A)):
                temp_list.append(A[row][column])
            trans.append(temp_list)        
        return(trans)

• 最简洁答案

class Solution:
    def transpose(self, A: List[List[int]]) -> List[List[int]]:
        return (list(zip(*A)))

• zip函数的介绍

zip() 函数用于将可迭代的对象作为参数，将对象中对应的元素打包成一个个元组，然后返回由这些元组组成的列表。
如果各个迭代器的元素个数不一致，则返回列表长度与最短的对象相同，利用 * 号操作符，可以将元组解压为列表。

>>>a = [1,2,3]
>>> b = [4,5,6]
>>> c = [4,5,6,7,8]
>>> zipped = zip(a,b)     # 打包为元组的列表
[(1, 4), (2, 5), (3, 6)]
>>> zip(a,c)              # 元素个数与最短的列表一致
[(1, 4), (2, 5), (3, 6)]
>>> zip(*zipped)          # 与 zip 相反，*zipped 可理解为解压，返回二维矩阵式
[(1, 2, 3), (4, 5, 6)]

• 补充
  星号‘*’的使用