Tôi cần chọn một số yếu tố từ danh sách nhất định, biết chỉ số của chúng. Giả sử tôi muốn tạo một danh sách mới, chứa phần tử có chỉ số 1, 2, 5, từ danh sách đã cho [-2, 1, 5, 3, 8, 5, 6]. Những gì tôi đã làm là:

a = [-2,1,5,3,8,5,6]
b = [1,2,5]
c = [ a[i] for i in b]

Có cách nào tốt hơn để làm điều đó? một cái gì đó như c = a [b]?

Bạn có thể sử dụng operator.itemgetter:

from operator import itemgetter 
a = [-2, 1, 5, 3, 8, 5, 6]
b = [1, 2, 5]
print(itemgetter(*b)(a))
# Result:
(1, 5, 5)

Hoặc bạn có thể sử dụng numpy :

import numpy as np
a = np.array([-2, 1, 5, 3, 8, 5, 6])
b = [1, 2, 5]
print(list(a[b]))
# Result:
[1, 5, 5]

Nhưng thực sự, giải pháp hiện tại của bạn là tốt. Nó có lẽ là gọn gàng nhất trong số họ.

Lựa chọn thay thế:

>>> map(a.__getitem__, b)
[1, 5, 5]

>>> import operator
>>> operator.itemgetter(*b)(a)
(1, 5, 5)

Một giải pháp khác có thể thông qua Sê-ri gấu trúc:

import pandas as pd

a = pd.Series([-2, 1, 5, 3, 8, 5, 6])
b = [1, 2, 5]
c = a[b]

Sau đó, bạn có thể chuyển đổi c trở lại danh sách nếu bạn muốn:

c = list(c)

Thử nghiệm cơ bản và không rộng rãi so sánh thời gian thực hiện của năm câu trả lời được cung cấp:

def numpyIndexValues(a, b):
    na = np.array(a)
    nb = np.array(b)
    out = list(na[nb])
    return out

def mapIndexValues(a, b):
    out = map(a.__getitem__, b)
    return list(out)

def getIndexValues(a, b):
    out = operator.itemgetter(*b)(a)
    return out

def pythonLoopOverlap(a, b):
    c = [ a[i] for i in b]
    return c

multipleListItemValues = lambda searchList, ind: [searchList[i] for i in ind]

sử dụng đầu vào sau:

a = range(0, 10000000)
b = range(500, 500000)

Vòng lặp python đơn giản là nhanh nhất với lambda hoạt động gần một giây, mapIndexValues ​​và getIndexValues ​​luôn khá giống nhau với phương thức numpy chậm hơn đáng kể sau khi chuyển đổi danh sách sang mảng numpy. Nếu dữ liệu đã có trong numpy Index thì phương thức numpy IndexValues ​​bị loại bỏ. nhanh nhất

numpyIndexValues -> time:1.38940598 (when converted the lists to numpy arrays)
numpyIndexValues -> time:0.0193445 (using numpy array instead of python list as input, and conversion code removed)
mapIndexValues -> time:0.06477512099999999
getIndexValues -> time:0.06391049500000001
multipleListItemValues -> time:0.043773591
pythonLoopOverlap -> time:0.043021754999999995

Tôi chắc chắn rằng điều này đã được xem xét: Nếu số lượng chỉ số trong b nhỏ và không đổi, người ta có thể viết kết quả như sau:

c = [a[b[0]]] + [a[b[1]]] + [a[b[2]]]

Hoặc thậm chí đơn giản hơn nếu chính các chỉ số là hằng số ...

c = [a[1]] + [a[2]] + [a[5]]

Hoặc nếu có một loạt các chỉ số liên tiếp ...

c = a[1:3] + [a[5]]

Đây là một cách đơn giản hơn:

a = [-2,1,5,3,8,5,6]
b = [1,2,5]
c = [e for i, e in enumerate(a) if i in b]

Câu trả lời của tôi không sử dụng bộ sưu tập numpy hoặc python.

Một cách tầm thường để tìm các yếu tố sẽ như sau:

a = [-2, 1, 5, 3, 8, 5, 6]
b = [1, 2, 5]
c = [i for i in a if i in b]

Nhược điểm: Phương pháp này có thể không hoạt động cho danh sách lớn hơn. Sử dụng numpy được khuyến khích cho danh sách lớn hơn.

Chỉ số tĩnh và danh sách nhỏ?

Đừng quên rằng nếu danh sách nhỏ và các chỉ mục không thay đổi, như trong ví dụ của bạn, đôi khi điều tốt nhất là sử dụng giải nén chuỗi :

_,a1,a2,_,_,a3,_ = a

Hiệu suất tốt hơn nhiều và bạn cũng có thể lưu một dòng mã:

 %timeit _,a1,b1,_,_,c1,_ = a
10000000 loops, best of 3: 154 ns per loop 
%timeit itemgetter(*b)(a)
1000000 loops, best of 3: 753 ns per loop
 %timeit [ a[i] for i in b]
1000000 loops, best of 3: 777 ns per loop
 %timeit map(a.__getitem__, b)
1000000 loops, best of 3: 1.42 µs per loop

Loại cách pythonic:

c = [x for x in a if a.index(x) in b]