Làm thế nào để trích xuất các phần tử thứ n từ danh sách các bộ dữ liệu trong python?

Question

Tôi đang cố gắng lấy các phần tử thứ n từ danh sách các bộ dữ liệu.

Tôi có một cái gì đó như:

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

tôi muốn trích xuất chỉ các yếu tố thứ hai của mỗi tuple vào một danh sách:

seconds = [1, 3, 5] 

Tôi biết rằng nó có thể được thực hiện với một vòng lặp for nhưng Tôi muốn biết nếu có một cách khác vì tôi có hàng ngàn bộ dữ liệu.

Answer 1

[x[1] for x in elements] 

Answer 2

Tôi biết rằng nó có thể được thực hiện với một CHO nhưng tôi muốn biết nếu có một cách khác

Có một cách khác. Bạn cũng có thể làm điều đó với map và itemgetter:

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

này vẫn thực hiện một vòng lặp trong nội bộ và mặc dù nó là hơi chậm hơn so với danh sách hiểu:

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))) 

Kết quả:

 
Method 1: 1.25699996948 
Method 2: 1.46600008011 

Nếu bạn cần lặp qua danh sách thì sử dụng for là tốt.

Answer 3

này cũng hoạt động:

zip(*elements)[1] 

(tôi chủ yếu là gửi bài này, để chứng minh cho bản thân mình mà tôi đã groked zip ...)

Nhìn thấy nó trong hành động:

>>> help(zip) 

Trợ giúp về chức năng zip tích hợp trong mô-đun builtin:

zip (...)

zip (seq1 [, seq2 [...]]) -> [(seq1 [0], seq2 [0] ...), (...) ]

Trả về danh sách các bộ dữ liệu, trong đó mỗi bộ chứa phần tử thứ i từ mỗi chuỗi đối số. Danh sách được trả lại bị cắt ngắn về độ dài của chuỗi đối số ngắn nhất.

>>> 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) 
>>> 

điều Neat Tôi đã học được ngày hôm nay: Sử dụng *list trong lập luận để tạo ra một danh sách tham số cho một hàm ...

Answer 4

Các mặt hàng này khi tôi đang tìm kiếm đó là cách nhanh nhất để kéo phần tử thứ hai của danh sách 2 tuple.Không phải những gì tôi muốn nhưng chạy cùng một thử nghiệm như hình với một phương pháp thứ 3 cộng với thử nghiệm phương pháp zip

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 

Vì vậy, qua nhanh gấp hai lần nếu bạn có một cặp 2 tuple chỉ chuyển đổi sang một dict và mất các giá trị.

Answer 5

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

Answer 6

Thời gian cho Python 3.6 để trích xuất phần tử thứ hai từ một danh sách 2-tuple.

Ngoài ra, thêm numpy phương thức mảng, đơn giản hơn để đọc (nhưng được cho là đơn giản hơn so với danh sách hiểu).

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] 

và timings:

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 

Lưu ý rằng map() và zip() không trả về một danh sách nữa, do đó việc chuyển đổi rõ ràng.