Convert 1D Array Into 2D Array LeetCode solution in Python

1
class Solution:
2
    def construct2DArray(self, original: List[int], m: int, n: int) -> List[List[int]]:
3
        ans = []
4
        if len(original) == m * n:
5
            for i in range(0, len(original), n):
6
                ans.append(original[i : i + n])
7
        return ans