python二维数组实现

问题：实现图片的排列 width代表列，height代表行
实现横向输出 和纵向输出
求源码

import re
inp = input()
data = re.findall(r"\d+.?\d*",inp)
width = int(data[0].replace(',',''))
height = int(data[1].replace(',',''))
maxnum = int(data[2].replace(',',''))

print('-'*30)

未完待续

import numpy as np
def cal_arry(width,heigth,maxNum):
    Matr=np.zeros(shape=(heigth,width))
    k=maxNum-1
    for i in range(heigth):
        for j in range(width):
            Matr[i,j]=maxNum-k
            k=k-1
            if k<0:
                break
        if k<0:
            break

    return Matr

result=cal_arry(4,3,5)
print(result)

输出结果：

[[1. 2. 3. 4.]
 [5. 0. 0. 0.]
 [0. 0. 0. 0.]]