จะแยกองค์ประกอบที่ n ออกจากรายการสิ่งที่เพิ่มขึ้นได้อย่างไร?

ฉันกำลังพยายามหาองค์ประกอบที่ n จากรายการสิ่งที่เพิ่มขึ้น

ฉันมีสิ่งที่ชอบ:

elements = [(1,1,1),(2,3,7),(3,5,10)]

ฉันต้องการแยกเฉพาะองค์ประกอบที่สองของแต่ละทูเปิลลงในรายการ:

seconds = [1, 3, 5]

ฉันรู้ว่ามันสามารถทำได้ด้วยการforวนซ้ำ แต่ฉันอยากรู้ว่ามีวิธีอื่นหรือไม่เนื่องจากฉันมีตัวทับหลายพันตัว

n = 1 # N. . .
[x[n] for x in elements]

สิ่งนี้ยังใช้งานได้:

zip(*elements)[1]

(ฉันโพสต์สิ่งนี้เป็นหลักเพื่อพิสูจน์ตัวเองว่าฉันได้บ่นzip... )

ดูการดำเนินการ:

>>> help(zip)

ความช่วยเหลือเกี่ยวกับฟังก์ชัน zip ในตัวในโมดูลที่สร้างขึ้น:

zip ( ... )

ซิป (seq1 [, seq2 [... ]]) -> [(seq1 [0], seq2 [0] ... ), (... )]

ส่งคืนรายการสิ่งทอโดยแต่ละทูเพิลมีองค์ประกอบ i-th จากแต่ละลำดับอาร์กิวเมนต์ รายการที่ส่งคืนจะถูกตัดทอนตามความยาวของความยาวของลำดับอาร์กิวเมนต์ที่สั้นที่สุด

>>> elements = [(1,1,1),(2,3,7),(3,5,10)]
>>> zip(*elements)
[(1, 2, 3), (1, 3, 5), (1, 7, 10)]
>>> zip(*elements)[1]
(1, 3, 5)
>>>

สิ่งที่ฉันเรียนรู้ในวันนี้: ใช้*listในอาร์กิวเมนต์เพื่อสร้างรายการพารามิเตอร์สำหรับฟังก์ชัน ...

หมายเหตุ : ใน Python3 zipส่งคืนตัววนซ้ำดังนั้นให้ใช้list(zip(*elements))เพื่อส่งคืนรายการสิ่งที่เพิ่มขึ้นแทน

ฉันรู้ว่ามันสามารถทำได้ด้วย FOR แต่ฉันอยากรู้ว่ามีวิธีอื่นอีกไหม

ยังมีอีกวิธีหนึ่ง คุณสามารถทำได้ด้วยแผนที่และตัวรับไอเทม :

>>> from operator import itemgetter
>>> map(itemgetter(1), elements)

สิ่งนี้ยังคงดำเนินการวนซ้ำภายในแม้ว่าจะช้ากว่าความเข้าใจรายการเล็กน้อย:

setup = 'elements = [(1,1,1) for _ in range(100000)];from operator import itemgetter'
method1 = '[x[1] for x in elements]'
method2 = 'map(itemgetter(1), elements)'

import timeit
t = timeit.Timer(method1, setup)
print('Method 1: ' + str(t.timeit(100)))
t = timeit.Timer(method2, setup)
print('Method 2: ' + str(t.timeit(100)))

ผล:

วิธีที่ 1: 1.25699996948
วิธีที่ 2: 1.46600008011

หากคุณต้องการทำซ้ำในรายการการใช้ a forก็ใช้ได้

พบสิ่งนี้ขณะที่ฉันกำลังค้นหาวิธีที่เร็วที่สุดในการดึงองค์ประกอบที่สองของรายการ 2 ทูเปิล ไม่ใช่สิ่งที่ฉันต้องการ แต่ใช้การทดสอบเดียวกันตามที่แสดงด้วยวิธีที่ 3 และทดสอบวิธีซิป

setup = 'elements = [(1,1) for _ in range(100000)];from operator import itemgetter'
method1 = '[x[1] for x in elements]'
method2 = 'map(itemgetter(1), elements)'
method3 = 'dict(elements).values()'
method4 = 'zip(*elements)[1]'

import timeit
t = timeit.Timer(method1, setup)
print('Method 1: ' + str(t.timeit(100)))
t = timeit.Timer(method2, setup)
print('Method 2: ' + str(t.timeit(100)))
t = timeit.Timer(method3, setup)
print('Method 3: ' + str(t.timeit(100)))
t = timeit.Timer(method4, setup)
print('Method 4: ' + str(t.timeit(100)))

Method 1: 0.618785858154
Method 2: 0.711684942245
Method 3: 0.298138141632
Method 4: 1.32586884499

เร็วกว่าสองเท่าถ้าคุณมีทูเพิล 2 คู่เพื่อแปลงเป็น dict แล้วรับค่า

การกำหนดเวลาสำหรับ Python 3.6สำหรับการแยกองค์ประกอบที่สองจากรายการ 2 ทูเพิล

นอกจากนี้ยังเพิ่มnumpyเมธอดอาร์เรย์ซึ่งง่ายกว่าในการอ่าน (แต่เนื้อหาง่ายกว่าความเข้าใจในรายการ)

from operator import itemgetter
elements = [(1,1) for _ in range(100000)]

%timeit second = [x[1] for x in elements]
%timeit second = list(map(itemgetter(1), elements))
%timeit second = dict(elements).values()
%timeit second = list(zip(*elements))[1]
%timeit second = np.array(elements)[:,1]

และการกำหนดเวลา:

list comprehension:  4.73 ms ± 206 µs per loop
list(map):           5.3 ms ± 167 µs per loop
dict:                2.25 ms ± 103 µs per loop
list(zip)            5.2 ms ± 252 µs per loop
numpy array:        28.7 ms ± 1.88 ms per loop

โปรดทราบว่าmap()และzip()อย่าส่งคืนรายการอีกต่อไปดังนั้นการแปลงอย่างชัดเจน

map (lambda x:(x[1]),elements)

การใช้isliceและchain.from_iterable:

>>> from itertools import chain, islice
>>> elements = [(1,1,1),(2,3,7),(3,5,10)]
>>> list(chain.from_iterable(islice(item, 1, 2) for item in elements))
[1, 3, 5]

สิ่งนี้จะมีประโยชน์เมื่อคุณต้องการมากกว่าหนึ่งองค์ประกอบ:

>>> elements = [(0, 1, 2, 3, 4, 5), 
                (10, 11, 12, 13, 14, 15), 
                (20, 21, 22, 23, 24, 25)]
>>> list(chain.from_iterable(islice(tuple_, 2, 5) for tuple_ in elements))
[2, 3, 4, 12, 13, 14, 22, 23, 24]