วิธีรับองค์ประกอบที่ n ของรายการไพ ธ อนหรือค่าดีฟอลต์หากไม่มี

ฉันกำลังมองหารายการเทียบเท่าใน python ของdictionary.get(key, default)สำหรับรายการ มีสำนวนหนึ่งซับในการรับองค์ประกอบที่ n ของรายการหรือค่าเริ่มต้นหากไม่สามารถใช้ได้?

ตัวอย่างเช่นรับรายการ myList ฉันต้องการได้รับmyList[0]หรือ 5 ถ้าmyListเป็นรายการว่างเปล่า

ขอบคุณ

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

เพื่อสนับสนุนดัชนีลบเราสามารถใช้:

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

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

แก้ไข: ฉันลบเช็คสำหรับ TypeError - น่าจะดีกว่าเพื่อให้ผู้เรียกจัดการสิ่งนี้

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

นี่อาจจะดีกว่าเมื่อโตaขึ้น

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

งานนี้เพราะถ้าa[n:]เป็นรายการที่ว่างเปล่าถ้าn => len(a)

นี่คือตัวอย่างของการทำงานกับ range(5)

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

และแสดงออกอย่างเต็มที่

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

เพิ่งค้นพบว่า:

next(iter(myList), 5)

iter(l)ส่งกลับ iterator บนmyList, next()กินองค์ประกอบแรกของ iterator และยกStopIterationข้อผิดพลาด แต่ถ้าเรียกว่ามีค่าเริ่มต้นซึ่งเป็นกรณีที่นี่อาร์กิวเมนต์ที่สอง5

ใช้งานได้เฉพาะเมื่อคุณต้องการองค์ประกอบที่ 1 ซึ่งเป็นกรณีในตัวอย่างของคุณ แต่ไม่ใช่ในคำถามที่คุณถามดังนั้น ...

นอกจากนี้มันไม่จำเป็นต้องสร้างรายการชั่วคราวในหน่วยความจำและทำงานได้ทุกประเภท iterable แม้ว่าจะไม่มีชื่อ (ดูความเห็นของ Xiong Chiamiov ในคำตอบของ gruszczy)

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

... กำลังมองหารายการเทียบเท่าใน python ของdict.get(key, default)สำหรับรายการ

มีสูตร itertoolsที่ทำเพื่อ iterables ทั่วไป เพื่อความสะดวกคุณสามารถ> pip install more_itertoolsและนำเข้าห้องสมุดบุคคลที่สามซึ่งใช้สูตรดังกล่าวสำหรับคุณ:

รหัส

import more_itertools as mit


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

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

รายละเอียด

นี่คือการดำเนินการตามnthสูตร:

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

เช่นเดียวกับdict.get()เครื่องมือนี้จะส่งคืนค่าเริ่มต้นสำหรับดัชนีที่หายไป มันใช้กับ iterables ทั่วไป:

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  

วิธีแก้ปัญหาราคาถูกคือการทำ dict ด้วยการแจกแจงและใช้.get()ตามปกติเช่น

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

เมื่อรวม @ Joachim's เข้ากับด้านบนคุณสามารถใช้

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

ตัวอย่าง:

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

หรืออาจชัดเจนมากขึ้น แต่ไม่มี len

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

ใช้ Python 3.4 ของcontextlib.suppress(exceptions)การสร้างวิธีการที่คล้ายกันgetitem()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