Cách lấy phần tử thứ n của danh sách python hoặc mặc định nếu không có sẵn

Tôi đang tìm kiếm một tương đương trong python của dictionary.get(key, default)danh sách. Có một thành ngữ lót nào để lấy phần tử thứ n của danh sách hoặc giá trị mặc định nếu không có sẵn không?

Ví dụ, đưa ra một danh sách myList tôi muốn nhận myList[0], hoặc 5 nếu myListlà một danh sách trống.

Cảm ơn.

l[index] if index < len(l) else default

Để hỗ trợ các chỉ số tiêu cực, chúng ta có thể sử dụng:

l[index] if -len(l) <= index < len(l) else default

try:
   a = b[n]
except IndexError:
   a = default

Chỉnh sửa: Tôi đã xóa kiểm tra cho TypeError - có lẽ tốt hơn để cho người gọi xử lý việc này.

(a[n:]+[default])[0]

Điều này có lẽ tốt hơn khi alớn hơn

(a[n:n+1]+[default])[0]

Điều này hoạt động bởi vì nếu a[n:]là một danh sách trống nếun => len(a)

Dưới đây là một ví dụ về cách làm việc này với range(5)

>>> range(5)[3:4]
[3]
>>> range(5)[4:5]
[4]
>>> range(5)[5:6]
[]
>>> range(5)[6:7]
[]

Và biểu hiện đầy đủ

>>> (range(5)[3:4]+[999])[0]
3
>>> (range(5)[4:5]+[999])[0]
4
>>> (range(5)[5:6]+[999])[0]
999
>>> (range(5)[6:7]+[999])[0]
999

Chỉ cần phát hiện ra rằng:

next(iter(myList), 5)

iter(l)trả về một iterator trên myList, next()sử dụng phần tử đầu tiên của iterator và gây ra StopIterationlỗi trừ khi được gọi với một giá trị mặc định, đó là trường hợp ở đây, đối số thứ hai,5

Điều này chỉ hoạt động khi bạn muốn phần tử thứ 1, đó là trường hợp trong ví dụ của bạn, nhưng không phải trong văn bản câu hỏi của bạn, vì vậy ...

Ngoài ra, nó không cần tạo danh sách tạm thời trong bộ nhớ và nó hoạt động cho bất kỳ loại lặp nào, ngay cả khi nó không có tên (xem nhận xét của Xiong Chiamiov về câu trả lời của gruszczy)

(L[n:n+1] or [somedefault])[0]

... đang tìm kiếm một tương đương trong python của dict.get(key, default)danh sách

Có một công thức itertools thực hiện điều này cho các lần lặp chung. Để thuận tiện, bạn có thể > pip install more_itertoolsvà nhập thư viện bên thứ ba này thực hiện các công thức nấu ăn đó cho bạn:

Mã

import more_itertools as mit


mit.nth([1, 2, 3], 0)
# 1    

mit.nth([], 0, 5)
# 5    

Chi tiết

Dưới đây là việc thực hiện các nthcông thức:

def nth(iterable, n, default=None):
    "Returns the nth item or a default value"
    return next(itertools.islice(iterable, n, None), default)

Giống như dict.get(), công cụ này trả về một mặc định cho các chỉ số bị thiếu. Nó áp dụng cho các lần lặp chung:

mit.nth((0, 1, 2), 1)                                      # tuple
# 1

mit.nth(range(3), 1)                                       # range generator (py3)
# 1

mit.nth(iter([0, 1, 2]), 1)                                # list iterator 
# 1  

Một giải pháp rẻ tiền là thực sự tạo ra một lệnh với liệt kê và sử dụng .get()như bình thường, như

 dict(enumerate(l)).get(7, my_default)

Kết hợp @ Joachim với những điều trên, bạn có thể sử dụng

next(iter(my_list[index:]), default)

Ví dụ:

next(iter(range(10)[8:]), 11)
8
>>> next(iter(range(10)[12:]), 11)
11

Hoặc, có thể rõ ràng hơn, nhưng không có len

my_list[index] if my_list[index:] else default

Sử dụng Python 3.4 contextlib.suppress(exceptions)để xây dựng một getitem()phương thức tương tự getattr().

import contextlib

def getitem(iterable, index, default=None):
    """Return iterable[index] or default if IndexError is raised."""
    with contextlib.suppress(IndexError):
        return iterable[index]
    return default