给定一个数组arr和一个整数n,将数组arr分成长度为n的小数组,并形成一个n x n的平方矩阵。
代码示例:
def split_array(arr, n):
# 判断数组长度和n的大小是否匹配
if len(arr) != n*n:
return "Error: The length of array is not compatible with n"
# 分割数组
subarrays = []
for i in range(0, len(arr), n):
subarrays.append(arr[i:i+n])
# 构建矩阵
matrix = []
for i in range(n):
row = subarrays[i]
matrix.append(row)
return matrix
示例输入:
arr = [1,2,3,4,5,6,7,8,9]
n = 3
示例输出:
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
下一篇:把一个数组分割成等和的连续子数组