Прибавить список, умноженный на скаляр

@Omny · Регистрация: 17.09.2016

Author24 — интернет-сервис помощи студентам

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

Задача, в которой возник вопрос, подразумевает поиск клада по указаниям. Клад находится в точке tk, координаты считываем. Мы стоим в точке k = (0, 0). Далее читаем пары строк из направления и количества шагов и делаем соответствующий шаг, пока не придём к кладу.

Python

tk = [int(input()), int(input())]
dirs = {'север': (0, 1), 'восток': (1, 0), 'юг': (0, -1), 'запад': (-1, 0)}
k = [0, 0]
steps = 0
while not (tk == k):
    steps += 1
    cur_dir = input()
    dist = int(input())
    k[0] += dirs[cur_dir][0]*dist
    k[1] += dirs[cur_dir][1]*dist
print(steps)

Пример входных данных:

Код

-5
1
север
1
запад
5
восток
7

Собственно вопрос про вот эти две строки:

Python

1 2	k[0] += dirs[cur_dir][0]dist k[1] += dirs[cur_dir][1]dist

Хотелось бы улучшить их вид, не ухудшив смысл\производительность\и т. д.

Думаю в сторону чего-то такого:

Python

1	k = [x + y*dist for x, y in zip(k, dirs[cur_dir])]

Но как-то громоздко выходит.

@dondublon · 19.09.2016, 11:19

На numpy такие вещи хорошо делаются.
В PyQt тоже есть подходящие типы.

@dondublon 4632 / 2050 / 361 Регистрация: 17.03.2012 Сообщений: 10,133 Записей в блоге: 6
	19.09.2016, 11:19	2
	На numpy такие вещи хорошо делаются. В PyQt тоже есть подходящие типы. 0