Hasilkan daftar angka dan pasangan negatifnya dengan Python

Apakah ada satu-liner yang nyaman untuk menghasilkan daftar angka dan rekan negatifnya di Python?

Misalnya, saya ingin membuat daftar dengan angka 6 hingga 9 dan -6 hingga -9.

Pendekatan saya saat ini adalah:

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

"Satu garis" sederhana adalah:

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

Namun, menghasilkan dua daftar dan kemudian bergabung bersama tampaknya tidak nyaman.

Saya tidak yakin jika pesanan itu penting, tetapi Anda bisa membuat tuple dan membukanya dalam pemahaman daftar.

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

Buat fungsi yang bagus dan mudah dibaca:

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

Pemakaian:

list(range_with_negatives(6, 10))

Itulah cara Anda mendapatkan satu-liner yang nyaman untuk apa pun. Hindari berusaha terlihat seperti hacker pro sihir.

Saya akan mengatakan solusi paling sederhana adalah membongkar dua rentang ke dalam daftar menggunakan *operator membongkar:

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

Tidak hanya ini jawaban terpendek yang diajukan, tetapi juga yang paling berkinerja, karena hanya membangun satu daftar dan tidak melibatkan pemanggilan fungsi selain keduanya. range .

Saya memverifikasi ini dengan menguji semua jawaban pertanyaan ini menggunakan timeit modul:

Jawaban ID Metode timeit hasilnya
-------------------------------------------------- ------------------------------------------------
(dalam pertanyaan) [x untuk x dalam kisaran (6,10)] + [y untuk y dalam kisaran (-9, -5)] 0,843 usec per loop
(jawaban ini) [* range (6, 10), * range (-9, -5)] 0,509 usec per loop
61348876 [y untuk x dalam kisaran (6,10) untuk y dalam (x, -x)] 0,754 usec per loop
61349149 daftar (range_with_negatives (6, 10)) 0,795 usec per loop
61348914 daftar (itertools.chain (kisaran (6, 10), kisaran (-9, -5))) 0,709 usec per loop
61366995 [tanda * x untuk tanda, x di itertools.product ((- 1, 1), range (6, 10))] 0,899 usec per loop
61371302 daftar (kisaran (6, 10)) + daftar (kisaran (-9, -5)) 0,729 usec per loop
61367180 daftar (range_with_negs (6, 10)) 1.95 usec per loop

^{(pengujian timeit dilakukan dengan Python 3.6.9 di komputer saya sendiri (spesifikasi rata-rata))}

Anda dapat menggunakan itertools.chain()untuk menggabungkan dua rentang.

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

Anda dapat menggunakan itertools.product, yang merupakan produk kartesius.

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

Ini mungkin lebih lambat karena Anda menggunakan perkalian, tetapi harus mudah dibaca.

Jika Anda menginginkan solusi yang berfungsi murni, Anda juga dapat mengimpor itertools.starmapdan operator.mul:

from itertools import product, starmap
from operator import mul

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

Namun, ini kurang dapat dibaca.

Anda benar-benar dekat, dengan menggabungkan dua rangeobjek. Tetapi ada cara yang lebih mudah untuk melakukannya:

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

Artinya, konversikan setiap rangeobjek ke daftar, dan kemudian gabungkan dua daftar.

Pendekatan lain, menggunakan itertools:

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

itertools.chain()seperti generalisasi +: alih-alih menambahkan dua daftar, ia mengaitkan satu iterator demi satu untuk membuat "super-iterator". Kemudian berikan itu ke list()dan Anda mendapatkan daftar konkret, dengan semua angka yang Anda inginkan dalam memori.

Menimbang dengan kemungkinan lain.

Jika Anda ingin keterbacaan satu-liner asli Anda cukup bagus, tapi saya akan mengubah rentang menjadi sama karena saya pikir batas negatif membuat segalanya menjadi kurang jelas.

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

IMO pendekatan menggunakan itertools.chain disajikan dalam beberapa jawaban lain jelas merupakan yang terbersih dari yang disediakan sejauh ini.

Namun, karena dalam kasus Anda urutan nilai tidak masalah , Anda dapat menghindari keharusan mendefinisikan dua rangeobjek eksplisit , dan dengan demikian menghindari melakukan semua off-by-one matematika yang diperlukan untuk rangepengindeksan negatif , dengan menggunakan 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]

Tad verbose, tetapi cukup mudah dibaca.

Pilihan lain yang serupa adalah menggunakan tuple / argumen membongkar dengan polos chain:

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

Lebih ringkas, tetapi saya menemukan tuple membongkar lebih sulit untuk grok dalam pemindaian cepat.

Ini adalah variasi pada tema (lihat @Derte Trdelnik 's jawaban ) mengikuti filosofi itertoolsdi mana

blok bangunan iterator [...] berguna sendiri atau dalam kombinasi.

Idenya adalah bahwa, sementara kita mendefinisikan fungsi baru, kita mungkin membuatnya secara umum:

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

dan menerapkannya pada rangeiterator tertentu :

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

Jika Anda ingin menjaga urutan yang Anda tentukan, Anda dapat menggunakan generator rentang built-in Python dengan syarat:

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

Yang memberi Anda output:

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

Dan juga bekerja dengan 0 sebagai awal untuk rentang penuh.

Mungkin ada berbagai cara untuk menyelesaikan pekerjaan.

Variabel yang diberikan: 1. mulai = 6 2. berhenti = 10

Anda dapat mencoba ini juga, untuk pendekatan yang berbeda:

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)

Aku suka simetri.

a = 6 b = 10

nums = [x + y untuk x in (- (a + b-1), 0) untuk y dalam rentang (a, b)]

Hasilnya harus -9, -8, ..., 8, 9.

Saya percaya bahwa ekspresi nums dapat ditingkatkan, yang mengikuti "dalam" dan "kisaran" masih terlihat tidak seimbang bagi saya.