1次元配列から２次元配列へ変換する関数

array_to_2dimention_array.py

def to_2dimentional_list(arr, row_no, col_no):
  res = []
  offset = 0
  for row in range(0, row_no):
    res.append(arr[offset:col_no+offset])
    offset = offset + col_no
  return res


a = [1,2,3,4,5,6]

b = to_2dimentional_list(a, 2, 3)
print(b) # [[1, 2, 3], [4, 5, 6]]

c = to_2dimentional_list(a, 3, 2)
print(c) # [[1, 2], [3, 4], [5, 6]]