LeetCode566

最新推荐文章于 2022-10-19 20:14:18 发布

斯德哥尔摩人

最新推荐文章于 2022-10-19 20:14:18 发布

阅读量141

点赞数

分类专栏： LeetCode 文章标签： LeetCode python list_reshape 切片 list_array

本文链接：https://blog.csdn.net/zuocuomiao/article/details/82217427

版权

LeetCode 专栏收录该内容

45 篇文章 0 订阅

订阅专栏

本题注意点：list平整化+切片技巧实现reshape，或者借助numpy.array投机取巧，但是较慢。

def matrixReshape(nums, r, c):
    """
    :type nums: List[List[int]]
    :type r: int
    :type c: int
    :rtype: List[List[int]]
    """
    if len(nums)*len(nums[0]) != r*c:
        return nums
    flatten = [item for i in nums for item in i]
    return [flatten[i*c : (i+1)*c] for i in range(r)]

    # version2（我）
    # import numpy as np
    # return np.array(nums).reshape(r, c).tolist()

# sample:
nums =[[1,2],
    [3,4]]
r, c = 1, 4
print(matrixReshape(nums, r, c))