关于Python二维列表的一点操作方法

Python二维列表中 各个对应元素进行数学运算

代码如下
进行加法运算

GK = [[1, 2, 3], [4, 5, 6, 7, 8, 9]]
G0 = [[2, 3, 4], [5, 6, 7, 8, 9, 10]]


def func(p, q):                                     # 函数形式
    h00, h0, g00, g0 = [], [], [], []
    for i in p:                                     # 将二维列表化成一维列表
        for j in i:
            h0.append(j)

    for a in q:
        for b in a:
            g0.append(b)
    # print(h0, g0)

    for i in range(0, min(len(h0), len(g0))):       # 二维列表相加   最终取短列表形式
        h00.append(h0[i] + g0[i])
    # print(h00)

    for i in range(0, len(h00)+1):
        if i == len(p[0]):
            g00.append(h00[i - len(p[0]):i])
        if i == len(p[1]) + len(p[0]):
            g00.append(h00[i - len(p[1]):i])
    return g00


print(func(GK, G0))