Вопрос по алгоритму в списке?

Question

[Timebird]

Здравствуйте.
Вопрос небольшой: допустим имеется список:a = [12 35 49] и нужно задать для каждого числа диапазон на +- два числа. Ну, то есть, в результате получить:

a = [10 11 12 13 14    33 34 35 36 37    47 48 49 50 51]

(пробелы поставил, чтобы просто видно было). Так-то без проблем его сделать, но вот не доходит, как сделать так, чтобы можно было менять пределы задаваемых изменений. То есть, допустим, сделать не на 2 такта, а на 3, чтоб было:

a = [9 10 11 12 13 14 15    32 33 34 35 36 37 38    46 47 48 49 50 51 52]

и т.д. Можете подсказать, как такое реализовать?

Answer 1

[Alexej Simakov]

a = [12,35,49]
bracket = 3;
result = []
for x in a:
	for i in range(-bracket,bracket+1):
		result.append(x+i);
print result

[9, 10, 11, 12, 13, 14, 15, 32, 33, 34, 35, 36, 37, 38, 46, 47, 48, 49, 50, 51, 52]

а можно и так

items = [12,35,49]
bracket = 3
result = sum(list(map(lambda x: range(x-bracket,x+bracket+1), items)),[])
print result

[9, 10, 11, 12, 13, 14, 15, 32, 33, 34, 35, 36, 37, 38, 46, 47, 48, 49, 50, 51, 52]

Comments

[Timebird]

спасибо большое!

[longclaps]

При всём уважении, не учите людей дурному:
result = sum(list(map(lambda x: range(x - bracket, x + bracket + 1), items)), [])
- не работает в Python3
- имеет сложность O(N^2)
- просто неряшливо выглядит по сравнению с классическим
result = [i for x in items for i in range(x - bracket, x + bracket + 1)]

[Alexej Simakov]

longclaps: о, классно, спасибо. я этот вариант тоже видел, но так и не въехал как его читать. это вроде List Comprehensions называется? Для себя я бы конечно остановился на первом варианте, все синтактические выверты языка меня обычно смущают.

[longclaps]

Alexej Simakov: Да, List Comprehensions. Читается, как и первый вариант:
[
i # пока держим в уме
for x in items # внешний цикл
....for i in range(x - bracket, x + bracket + 1)] # внутренний цикл
........# вот теперь говорим "добавить в список i"
]

[Alexej Simakov]

longclaps: вроде понял, вот тут обьясняется совсем на пальцах python-list-comprehensions-now-in-color
то что мы добавляем в список пишем спереди, а после этого все выражения циклов и/или условий снаружи внутрь друг за другом. Вот, век живи век учись. Спасибо, а то так бы и обходил эту тему стороной.

[Влад]

longclaps: будто ваш вариант имеет иную сложность, бгг

[longclaps]

Влад: ? Обоснуй.