LeetCode(867)
题目描述
给定一个矩阵 A
, 返回 A
的转置矩阵。
矩阵的转置是指将矩阵的主对角线翻转,交换矩阵的行索引与列索引。
示例 1:
输入:[[1,2,3],[4,5,6],[7,8,9]]
输出:[[1,4,7],[2,5,8],[3,6,9]]
示例 2:
输入:[[1,2,3],[4,5,6]]
输出:[[1,4],[2,5],[3,6]]
提示:
1 <= A.length <= 1000
1 <= A[0].length <= 1000
解决方案
1.常规复制
class Solution:
def transpose(self, A):
"""
此方法的本质就是ans[c][r] = A[r][c]
:type A: List[List[int]]
:rtype: List[List[int]]
"""
# R——A的长度,C——数组的长度
R,C =len(A),len(A[0])
ans = [[None]*R for _ in range(C)]
# ans [[None, None, None], [None, None, None], [None, None, None]]
for r,row in enumerate(A):
for c,val in enumerate(row):
ans[c][r]= val
return ans
def main():
s =Solution()
b = s.transpose([[1,2,3],[4,5,6],[7,8,9]])
print(b)
#[[1, 4, 7], [2, 5, 8], [3, 6, 9]]
if __name__ == "__main__":
main()
2.zip函数一行搞定
class Solution:
def transpose(self, A):
"""
:type A: List[List[int]]
:rtype: List[List[int]]
"""
return list(zip(*A))
def main():
s =Solution()
b = s.transpose([[1,2,3],[4,5,6],[7,8,9]])
print(b)
#[[1, 4, 7], [2, 5, 8], [3, 6, 9]]
if __name__ == "__main__":
main()
补充:zip函数的用法
描述
zip() 函数用于将可迭代的对象作为参数,将对象中对应的元素打包成一个个元组,然后返回由这些元组组成的对象,这样做的好处是节约了不少的内存。
我们可以使用 list() 转换来输出列表。
如果各个迭代器的元素个数不一致,则返回列表长度与最短的对象相同,利用 * 号操作符,可以将元组解压为列表。
实例
>>>a = [1,2,3]
>>> b = [4,5,6]
>>> c = [4,5,6,7,8]
>>> zipped = zip(a,b) # 返回一个对象
>>> zipped
<zip object at 0x103abc288>
>>> list(zipped) # list() 转换为列表
[(1, 4), (2, 5), (3, 6)]
>>> list(zip(a,c)) # 元素个数与最短的列表一致
[(1, 4), (2, 5), (3, 6)]
>>> a1, a2 = zip(*zip(a,b)) # 与 zip 相反,zip(*) 可理解为解压,返回二维矩阵式
>>> list(a1)
[1, 2, 3]
>>> list(a2)
[4, 5, 6]
>>>
本文来自博客园,作者:YanceDev,转载请注明原文链接:https://www.cnblogs.com/yance-dev/p/10258838.html