Как добавить номер к каждому элементу в списке a, который дает новый список b с суммой добавленного номера

Question

Привет. Как добавить число к каждому целому числу в списке A, которое генерирует новый список B, используя python. Также я хочу сгенерировать n списков. Пример: a = [1,2,3] и n = 2 добавить k = 3 к каждому элементу в a, который дает b = [4,5,6], и добавить k в список b. что дает c = [7,8,9] Я достиг этого, используя функцию карты, но я хочу сгенерировать n списков.

def bw(n, k):
    test_list = [1,2,3]
    if n > 1:
        res = list(map(lambda x :x+k, test_list))
        return test_list, res
    else:
        return test_list
print bw(3, 2)

Выход b = [4,5,6] c = [7, 8,9]

Answer 1

Вот решение:

test_list = [1,2,3]

def bw(test_list, n, k):
    output= []
    temp = test_list
    for i in range(n):
        temp = [j + k for j in temp]
        output.append(temp)
    return output

print (bw(test_list, 2, 3))

Answer 2

Если вам нужен только окончательный массив:

def bw(n, k):
    test_list = [1,2,3]
    return [x+n*k for x in test_list]

print(bw(3, 2))

Если вам нужен также список списков:

def bw(n, k):
    test_list = [1,2,3]
    res = []
    for i in range(0,n):
        test_list = [x+k for x in test_list]
        res.append(test_list)

    return test_list, res

print(bw(3, 2))

Answer 3

res = [x + k for x in test_list]

Это прямо из GeeksforGeeks.