python chọn yếu tố cụ thể từ một danh sách

Question

Có một "pythonic" cách để nhận được giá trị nhất định từ một danh sách, tương tự như mã perl này:

my ($one,$four,$ten) = line.split(/,/)[1,4,10] 

Answer 1

Tôi nghĩ rằng bạn đang tìm kiếm operator.itemgetter:

import operator 
line=','.join(map(str,range(11))) 
print(line) 
# 0,1,2,3,4,5,6,7,8,9,10 
alist=line.split(',') 
print(alist) 
# ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '10'] 
one,four,ten=operator.itemgetter(1,4,10)(alist) 
print(one,four,ten) 
# ('1', '4', '10') 

Answer 2

lst = line.split(',') 
one, four, ten = lst[1], lst[4], lst[10] 

Answer 3

Có:

data = line.split(',') 
one, four, ten = data[1], data[4], data[10] 

Bạn cũng có thể sử dụng itemgetter, nhưng tôi thích mã ở trên, nó rõ ràng hơn, và rõ ràng == mã tốt.

Answer 4

Hãy thử operator.itemgetter (có sẵn trong python 2.4 hoặc mới hơn):

Return một đối tượng có thể được gọi là lấy về mục từ toán hạng của nó sử dụng phương pháp của toán hạng ____ GetItem ____(). Nếu nhiều mục được chỉ định, trả về một bộ giá trị tra cứu.

>>> from operator import itemgetter 
>>> line = ','.join(map(str, range(11))) 
>>> line 
'0,1,2,3,4,5,6,7,8,9,10' 
>>> a, b, c = itemgetter(1, 4, 10)(line.split(',')) 
>>> a, b, c 
('1', '4', '10') 

Condensed:

>>> # my ($one,$four,$ten) = line.split(/,/)[1,4,10] 
>>> from operator import itemgetter 
>>> (one, four, ten) = itemgetter(1, 4, 10)(line.split(',')) 

Answer 5

Sử dụng một danh sách hiểu

line = '0,1,2,3,4,5,6,7,8,9,10' 
lst = line.split(',') 
one, four, ten = [lst[i] for i in [1,4,10]] 

Answer 6

Làm thế nào về điều này:

index = [1, 0, 0, 1, 0] 
x = [1, 2, 3, 4, 5] 
[i for j, i in enumerate(x) if index[j] == 1] 
#[1, 4] 

Answer 7

Ngoài ra, nếu bạn có một mảng Numpy thay vì danh sách, bạn có thể làm một cái gì đó như:

from numpy import array 

# Assuming line = "0,1,2,3,4,5,6,7,8,9,10" 
line_array = array(line.split(",")) 

one, four, ten = line_array[[1,4,10]] 

Bí quyết ở đây là bạn có thể chuyển danh sách (hoặc mảng Numpy) làm chỉ mục mảng.

CHỈNH SỬA: Lần đầu tiên tôi nghĩ nó cũng sẽ hoạt động với bộ dữ liệu nhưng phức tạp hơn một chút. Tôi đề nghị gắn bó với danh sách hoặc mảng.