def matrix_union(A, B):
    for a, b in zip(A, B):
        yield [*a, *b]

A = [
    [0, 0, 1],
    [0, 1, 0],
    [1, 0, 1]
]

B = [
    [2, 2, 3],
    [3, 2, 2],
    [3, 3, 3]
]

print(list(matrix_union(A, B)))