Створення списку чисел та їх негативних аналогів у Python

Чи є зручний однолінійний параметр для створення списку чисел та їх негативних аналогів у Python?

Наприклад, скажіть, що я хочу створити список із числами 6 - 9 та -6 - -9.

Мій сучасний підхід:

l = [x for x in range(6,10)]
l += [-x for x in l]

Простий "однолінійний" буде:

l = [x for x in range(6,10)] + [y for y in range(-9, -5)]

Однак створювати два списки та потім об'єднувати їх разом здається незручним.

Я не впевнений, що замовлення має значення, але ви можете створити кортеж і розпакувати його в списку.

nums = [y for x in range(6,10) for y in (x,-x)]
print(nums)
[6, -6, 7, -7, 8, -8, 9, -9]

Створіть приємну та читану функцію:

def range_with_negatives(start, end):
    for x in range(start, end):
        yield x
        yield -x

Використання:

list(range_with_negatives(6, 10))

Ось так ви отримуєте зручну однолінійку для будь-чого. Уникайте спроб схожих на магічного про-хакера.

Я б сказав, що найпростішим рішенням є розпакування двох діапазонів у список за допомогою *оператора розпакування:

>>> [*range(6, 10), *range(-9, -5)]
[6, 7, 8, 9, -9, -8, -7, -6]

Це не лише найкоротший запропонований відповідь, але він також є найефективнішим, оскільки він лише створює єдиний список і не передбачає жодних викликів функцій за межами двох ranges.

Я перевірив це, перевіривши всі відповіді на це запитання за допомогою timeitмодуля:

Метод ідентифікатора відповіді timeit результат
-------------------------------------------------- ------------------------------------------------
(питання) [x для x у діапазоні (6,10)] + [y для y у діапазоні (-9, -5)] 0,843 Usec на цикл
(ця відповідь) [* діапазон (6, 10), * діапазон (-9, -5)] 0,509 Usec на цикл
61348876 [y для x у діапазоні (6,10) для y в (x, -x)] 0,754 Usec на цикл
61349149 список (range_with_negatives (6, 10)) 0,795 Usec на цикл
61348914 список (itertools.chain (діапазон (6, 10), діапазон (-9, -5))) 0,709 Usec за цикл
61366995 [знак * x для знака, x у itertools.product ((- 1, 1), діапазон (6, 10))] 0,899 Usec за цикл
61371302 список (діапазон (6, 10)) + список (діапазон (-9, -5)) 0,729 використання в циклі
61367180 список (range_with_negs (6, 10)) 1,95 Usec за цикл

^{(тестова перевірка, виконана з Python 3.6.9 на моєму комп’ютері (середні характеристики))}

Ви можете використовувати itertools.chain()для об'єднання двох діапазонів.

import itertools
list(itertools.chain(range(6, 10), range(-9, -5)))

Можна використовувати itertools.product, що є декартовим продуктом.

[sign*x for sign, x in product((-1, 1), range(6, 10))]
[-6, -7, -8, -9, 6, 7, 8, 9]

Це може бути повільніше, оскільки ви використовуєте множення, але його слід легко читати.

Якщо ви хочете суто функціональне рішення, ви також можете імпортувати itertools.starmapта operator.mul:

from itertools import product, starmap
from operator import mul

list(starmap(mul, product((-1, 1), range(6, 10))))

Однак це менш читабельно.

Ви дійсно близькі, поєднуючи два rangeоб’єкти. Але є простіший спосіб зробити це:

>>> list(range(6, 10)) + list(range(-9, -5))
[6, 7, 8, 9, -9, -8, -7, -6]

Тобто перетворіть кожен rangeоб’єкт у список, а потім з'єднайте два списки.

Ще один підхід із використанням itertools:

>>> list(itertools.chain(range(6, 10), range(-9, -5)))
[6, 7, 8, 9, -9, -8, -7, -6]

itertools.chain()це як узагальнений +: замість того, щоб додавати два списки, він ланцюгує один ітератор за іншим, щоб зробити "супер-ітератор". Потім передайте це, list()і ви отримаєте конкретний список із усіма цифрами, які потрібно в пам'яті.

Зважування є ще однією можливістю.

Якщо ви хочете читати, ваш оригінальний однолінійний лайнер був досить хорошим, але я би змінив діапазони, щоб вони були такими ж, як я думаю, що негативні межі роблять речі менш чіткими.

[x for x in range(6, 10)] + [-x for x in range(6, 10)]

ІМО, підхід із застосуванням itertools.chainпредставлених у кількох інших відповідях, безумовно, є найчистішим із запропонованих на сьогоднішній день.

Однак, оскільки у вашому випадку порядок значень не має значення , ви можете уникнути необхідності визначення двох явних rangeоб’єктів і, таким чином, уникнути проведення всієї математики, яка необхідна для негативного rangeіндексування, використовуючи itertools.chain.from_iterable:

>>> import itertools
>>> list(itertools.chain.from_iterable((x, -x) for x in range(6, 10)))
[6, -6, 7, -7, 8, -8, 9, -9]

Тад багатослівний, але досить читабельний.

Іншим подібним варіантом є використання розпакування кортежу / аргументу простою chain:

>>> list(itertools.chain(*((x, -x) for x in range(6, 10))))
[6, -6, 7, -7, 8, -8, 9, -9]

Більш лаконічний, але мені здається, що кортеж розпаковується важче, щоб швидко проскочити.

Це варіація на тему (див @Derte Trdelnik «s відповідь ) , слідуючи філософії , itertoolsде

Ітераторні будівельні блоки [...] корисні самі по собі або в поєднанні.

Ідея полягає в тому, що, визначаючи нову функцію, ми можемо зробити її загальною:

def interleaved_negatives(it):
    for i in it:
        yield i
        yield -i

і застосувати його до певного rangeітератора:

list(interleaved_negatives(range(6, 10)))

Якщо ви хочете зберегти вказаний вами порядок, ви можете скористатися вбудованим генератором діапазону Python із умовними умовами:

def range_with_negs(start, stop):
    for value in range(-(stop-1), stop):      
        if (value <= -start) or (value >= start):
            yield value

Що дає результат:

In [1]: list(range_with_negs(6, 10))
Out[1]: [-9, -8, -7, -6, 6, 7, 8, 9]

А також працює з 0 як початок для повного діапазону.

Існують різні способи виконання роботи.

Дані змінні: 1. старт = 6 2. стоп = 10

Ви можете спробувати це також для різних підходів:

def mirror_numbers(start,stop):
  if start<stop:
    val=range(start,stop)
    return [j if i < len(val) else -j for i,j in enumerate([x for x in val]*2) ]

mirror_numbers(6,10)

Я просто люблю симетрії.

a = 6 b = 10

nums = [x + y для x in (- (a + b-1), 0) для y в діапазоні (a, b)]

Результат повинен бути -9, -8, ..., 8, 9.

Я вважаю, що вираз nums можна вдосконалити, що далі "in" та "range" мені все ще виглядає незбалансованим.