Python idiom เพื่อส่งคืนไอเท็มแรกหรือไม่มี

ฉันแน่ใจว่ามีวิธีที่ง่ายกว่าในการทำเช่นนี้ซึ่งไม่ได้เกิดขึ้นกับฉัน

ฉันกำลังเรียกวิธีการมากมายที่ส่งคืนรายการ รายการอาจว่างเปล่า หากรายการไม่ว่างฉันต้องการส่งคืนรายการแรก; มิฉะนั้นฉันต้องการคืนค่าไม่มี รหัสนี้ใช้งานได้:

my_list = get_list()
if len(my_list) > 0: return my_list[0]
return None

สำหรับฉันแล้วดูเหมือนว่าควรมีสำนวนแบบบรรทัดเดียวที่เรียบง่ายสำหรับการทำสิ่งนี้ แต่สำหรับชีวิตของฉันฉันไม่สามารถคิดได้ มีอะไรบ้าง

แก้ไข:

เหตุผลที่ฉันกำลังมองหานิพจน์บรรทัดเดียวที่นี่ไม่ใช่ว่าฉันชอบรหัสตัวอักษรที่เหลือเชื่อ แต่เพราะฉันต้องเขียนโค้ดจำนวนมากเช่นนี้:

x = get_first_list()
if x:
    # do something with x[0]
    # inevitably forget the [0] part, and have a bug to fix
y = get_second_list()
if y:
    # do something with y[0]
    # inevitably forget the [0] part AGAIN, and have another bug to fix

สิ่งที่ฉันต้องการจะทำแน่นอนสามารถทำได้ด้วยฟังก์ชั่น (และอาจจะ):

def first_item(list_or_none):
    if list_or_none: return list_or_none[0]

x = first_item(get_first_list())
if x:
    # do something with x
y = first_item(get_second_list())
if y:
    # do something with y

ฉันโพสต์คำถามเพราะฉันประหลาดใจกับการแสดงออกอย่างง่าย ๆ ใน Python ที่สามารถทำได้และฉันคิดว่าการเขียนฟังก์ชั่นเป็นเรื่องโง่ที่ต้องทำหากมีการแสดงออกอย่างง่ายสามารถทำกลอุบายได้ แต่เมื่อเห็นคำตอบเหล่านี้ดูเหมือนว่าฟังก์ชั่นเป็นทางออกที่ง่าย

Python 2.6+

next(iter(your_list), None)

ถ้าyour_listสามารถNone:

next(iter(your_list or []), None)

Python 2.4

def get_first(iterable, default=None):
    if iterable:
        for item in iterable:
            return item
    return default

ตัวอย่าง:

x = get_first(get_first_list())
if x:
    ...
y = get_first(get_second_list())
if y:
    ...

ตัวเลือกอื่นคือการแทรกฟังก์ชั่นด้านบน:

for x in get_first_list() or []:
    # process x
    break # process at most one item
for y in get_second_list() or []:
    # process y
    break

เพื่อหลีกเลี่ยงbreakคุณสามารถเขียน:

for x in yield_first(get_first_list()):
    x # process x
for y in yield_first(get_second_list()):
    y # process y

ที่ไหน:

def yield_first(iterable):
    for item in iterable or []:
        yield item
        return

วิธีที่ดีที่สุดคือ:

a = get_list()
return a[0] if a else None

คุณสามารถทำได้ในหนึ่งบรรทัด แต่มันยากสำหรับโปรแกรมเมอร์ที่จะอ่าน:

return (get_list()[:1] or [None])[0]

(get_list() or [None])[0]

ว่าควรจะทำงาน

BTW ฉันไม่ได้ใช้ตัวแปรlistเพราะมันเขียนทับlist()ฟังก์ชั่นบิวอิน

แก้ไข: ฉันมีรุ่นที่เรียบง่ายกว่าเล็กน้อย แต่ไม่ถูกต้องที่นี่ก่อนหน้านี้

วิธีที่ใช้สำนวนภาษาไพ ธ อนมากที่สุดคือการใช้ตัวถัดไป () บนตัววนซ้ำเนื่องจากรายการสามารถทำซ้ำได้ เช่นเดียวกับสิ่งที่ @JFSebastian แสดงความคิดเห็นในวันที่ 13 ธันวาคม 2011

next(iter(the_list), None)ผลตอบแทนนี้ไม่มีถ้าthe_listเป็นที่ว่างเปล่า ดูถัดไป () Python 2.6+

หรือถ้าคุณรู้ว่าthe_listไม่ว่าง:

iter(the_list).next()ดูiterator.next () Python 2.2+

หากคุณพบว่าตัวเองพยายามถอนสิ่งแรก (หรือไม่มี) จากรายการความเข้าใจคุณสามารถสลับไปยังเครื่องกำเนิดไฟฟ้าเพื่อทำสิ่งที่ชอบ:

next((x for x in blah if cond), None)

Pro: ทำงานได้ถ้า blah ไม่สามารถจัดทำดัชนี Con: มันเป็นไวยากรณ์ที่ไม่คุ้นเคย มันมีประโยชน์ในขณะที่ทำการแฮ็กและกรองข้อมูลใน ipython

ทางออกของ OP ใกล้จะถึงแล้วมีเพียงไม่กี่อย่างเท่านั้นที่จะทำให้ Pythonic เป็นมากกว่า

สำหรับหนึ่งไม่จำเป็นต้องได้รับความยาวของรายการ รายการที่ว่างใน Python จะประเมินเป็น False ใน a if check เพียงแค่พูด

if list:

นอกจากนี้ยังเป็นแนวคิดที่ไม่ดีอย่างมากที่จะกำหนดให้กับตัวแปรที่ทับซ้อนกับคำที่สงวนไว้ "list" เป็นคำสงวนใน Python

งั้นลองเปลี่ยนมันเป็น

some_list = get_list()
if some_list:

จุดสำคัญจริงๆว่าจำนวนมากของการแก้ปัญหาที่นี่พลาดคือการที่ทุกฟังก์ชั่หลาม / วิธีกลับไม่มีโดยค่าเริ่มต้น ลองดังต่อไปนี้

def does_nothing():
    pass

foo = does_nothing()
print foo

ยกเว้นว่าคุณต้องการส่งคืน None เพื่อยกเลิกฟังก์ชั่นก่อน แต่ก็ไม่จำเป็นที่จะต้องส่งคืน None อย่างชัดเจน ค่อนข้างรัดกุมเพียงแค่ส่งคืนรายการแรกควรมีอยู่

some_list = get_list()
if some_list:
    return list[0]

และในที่สุดบางทีนี่อาจบอกเป็นนัย แต่เพียงเพื่อให้ชัดเจน (เพราะชัดแจ้งดีกว่าโดยปริยาย ) คุณไม่ควรให้ฟังก์ชันของคุณรับรายการจากฟังก์ชันอื่น เพียงส่งผ่านเป็นพารามิเตอร์ ดังนั้นผลลัพธ์สุดท้ายก็คือ

def get_first_item(some_list): 
    if some_list:
        return list[0]

my_list = get_list()
first_item = get_first_item(my_list)

อย่างที่ฉันบอกว่า OP ใกล้จะถึงแล้วและมีเพียงไม่กี่สัมผัสที่ให้รสชาติของ Python ที่คุณต้องการ

for item in get_list():
    return item

พูดตรงไปตรงมาฉันไม่คิดว่ามีสำนวนที่ดีกว่า: คุณชัดเจนและกระชับ - ไม่จำเป็นต้องมีอะไร "ดีกว่า" บางที แต่นี้เป็นจริงเรื่องของรสชาติที่คุณสามารถเปลี่ยนif len(list) > 0:ด้วยif list:- รายการที่ว่างเปล่าจะประเมินการเท็จ

ในบันทึกย่อที่เกี่ยวข้อง Python ไม่ใช่ Perl (ไม่มีการเล่นสำนวนเจตนา!) คุณไม่จำเป็นต้องได้รหัสที่เยี่ยมที่สุดเท่าที่จะเป็นไปได้
จริงๆแล้วรหัสที่แย่ที่สุดที่ฉันเคยเห็นใน Python ก็คือ :-) ยอดเยี่ยมมาก

โดยวิธีการแก้ปัญหาส่วนใหญ่ที่ฉันได้เห็นที่นี่ไม่ได้คำนึงถึงเมื่อรายการ [0] ประเมินเป็นเท็จ (เช่นสตริงว่างหรือศูนย์) - ในกรณีนี้พวกเขาทั้งหมดกลับไม่มีและไม่ใช่องค์ประกอบที่ถูกต้อง

Python idiom ที่จะคืนค่าไอเท็มแรกหรือไม่?

วิธีการไพ ธ อนิกที่มากที่สุดคือสิ่งที่คำตอบที่ตอบโต้ได้สูงที่สุดแสดงให้เห็นและมันเป็นสิ่งแรกที่เข้ามาในใจของฉันเมื่อฉันอ่านคำถาม ต่อไปนี้เป็นวิธีใช้งานก่อนอื่นถ้ารายการที่ว่างเปล่าอาจถูกส่งเข้าไปในฟังก์ชัน:

def get_first(l): 
    return l[0] if l else None

และถ้ารายการถูกส่งคืนจากget_listฟังก์ชัน:

l = get_list()
return l[0] if l else None

วิธีอื่นแสดงให้เห็นถึงการทำสิ่งนี้ที่นี่พร้อมคำอธิบาย

for

เมื่อฉันเริ่มพยายามคิดวิธีที่ฉลาดในการทำสิ่งนี้สิ่งที่สองที่ฉันคิดคือ

for item in get_list():
    return item

สิ่งนี้จะสันนิษฐานว่าฟังก์ชั่นสิ้นสุดที่นี่โดยปริยายคืนNoneถ้าget_listส่งกลับรายการที่ว่างเปล่า รหัสที่ชัดเจนด้านล่างเทียบเท่าอย่างแน่นอน:

for item in get_list():
    return item
return None

if some_list

ต่อไปนี้ยังได้รับการเสนอ (ฉันแก้ไขชื่อตัวแปรไม่ถูกต้อง) Noneซึ่งยังใช้โดยปริยาย สิ่งนี้จะดีกว่าที่กล่าวมาเนื่องจากใช้การตรวจสอบแบบลอจิคัลแทนการทำซ้ำที่อาจไม่เกิดขึ้น ควรเข้าใจได้ง่ายขึ้นในทันทีว่าเกิดอะไรขึ้น แต่ถ้าเราเขียนเพื่อความสามารถในการอ่านและการบำรุงรักษาเราควรเพิ่มความชัดเจนreturn Noneในตอนท้าย:

some_list = get_list()
if some_list:
    return some_list[0]

ฝานor [None]และเลือกดัชนีซีโรท

คำตอบนี้ได้รับการโหวตมากที่สุดด้วย:

return (get_list()[:1] or [None])[0]

Slice นั้นไม่จำเป็นและสร้างรายการพิเศษหนึ่งรายการในหน่วยความจำ ต่อไปนี้ควรมีประสิทธิภาพมากขึ้น ในการอธิบายให้orส่งคืนองค์ประกอบที่สองหากองค์ประกอบแรกอยู่Falseในบริบทบูลีนดังนั้นหากget_listส่งคืนรายการว่างเปล่านิพจน์ที่อยู่ในวงเล็บจะส่งคืนรายการด้วย 'ไม่มี' ซึ่งจะเข้าถึง0ดัชนีโดย:

return (get_list() or [None])[0]

รายการถัดไปใช้ข้อเท็จจริงและส่งคืนรายการที่สองหากรายการแรกอยู่Trueในบริบทบูลีนและเนื่องจากมันอ้างอิง my_list สองครั้งจึงไม่ดีไปกว่านิพจน์ประกอบไปด้วย (และในทางเทคนิคไม่ใช่ซับไลน์):

my_list = get_list() 
return (my_list and my_list[0]) or None

next

จากนั้นเรามีการใช้งานที่ชาญฉลาดของ builtin nextและiter

return next(iter(get_list()), None)

ในการอธิบายให้iterส่งคืนตัววนซ้ำด้วย.nextเมธอด ( .__next__ใน Python 3) จากนั้น builtin จะnextเรียก.nextใช้เมธอดนั้นและหากตัววนซ้ำหมดแล้วส่งคืนค่าเริ่มต้นที่เราNoneกำหนด

นิพจน์ประกอบไปด้วยซ้ำซ้อน ( a if b else c) และวนกลับ

มีการเสนอด้านล่าง แต่สิ่งที่ตรงกันข้ามควรจะดีกว่าเนื่องจากตรรกะมักจะเข้าใจดีกว่าในแง่บวกแทนที่จะเป็นเชิงลบ เนื่องจากget_listมีการเรียกสองครั้งยกเว้นว่ามีการบันทึกผลลัพธ์ในบางวิธีสิ่งนี้จะทำงานได้ไม่ดี:

return None if not get_list() else get_list()[0]

สิ่งที่ตรงกันข้ามที่ดีกว่า:

return get_list()[0] if get_list() else None

ยิ่งไปกว่านั้นให้ใช้ตัวแปรโลคอลเพื่อที่get_listจะเรียกได้เพียงครั้งเดียวเท่านั้นและคุณมีวิธีการแก้ปัญหาแบบ Pythonic ที่แนะนำครั้งแรก:

l = get_list()
return l[0] if l else None

เกี่ยวกับสำนวนมีสูตร itertoolsnthเรียกว่า

จากสูตร itertools:

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

หากคุณต้องการ one-liners ให้ลองติดตั้งไลบรารี่ที่ใช้สูตรนี้สำหรับคุณเช่นmore_itertools:

import more_itertools as mit

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

mit.nth([], 0)                                             # default is `None`
# None

มีเครื่องมืออีกชิ้นที่สามารถส่งคืนไอเท็มแรกmore_itertools.firstเท่านั้น

mit.first([3, 2, 1])
# 3

mit.first([], default=None)
# None

itertools เหล่านี้ขยายโดยทั่วไปสำหรับ iterable ใด ๆ ไม่เพียง แต่สำหรับรายการ

my_list[0] if len(my_list) else None

ฉันหมดเวลากับการแก้ปัญหาสองข้อด้วยความอยากรู้ วิธีแก้ปัญหาที่ใช้คำสั่ง return เพื่อสิ้นสุด a for loop ก่อนกำหนดนั้นมีค่าใช้จ่ายสูงกว่าเล็กน้อยในเครื่องของฉันด้วย Python 2.5.1 ฉันสงสัยว่านี่เกี่ยวข้องกับการตั้งค่า iterable

import random
import timeit

def index_first_item(some_list):
    if some_list:
        return some_list[0]


def return_first_item(some_list):
    for item in some_list:
        return item


empty_lists = []
for i in range(10000):
    empty_lists.append([])

assert empty_lists[0] is not empty_lists[1]

full_lists = []
for i in range(10000):
    full_lists.append(list([random.random() for i in range(10)]))

mixed_lists = empty_lists[:50000] + full_lists[:50000]
random.shuffle(mixed_lists)

if __name__ == '__main__':
    ENV = 'import firstitem'
    test_data = ('empty_lists', 'full_lists', 'mixed_lists')
    funcs = ('index_first_item', 'return_first_item')
    for data in test_data:
        print "%s:" % data
        for func in funcs:
            t = timeit.Timer('firstitem.%s(firstitem.%s)' % (
                func, data), ENV)
            times = t.repeat()
            avg_time = sum(times) / len(times)
            print "  %s:" % func
            for time in times:
                print "    %f seconds" % time
            print "    %f seconds avg." % avg_time

นี่คือเวลาที่ฉันได้รับ:

empty_lists:
  index_first_item:
    0.748353 วินาที
    0.741086 วินาที
    0.741191 วินาที
    เฉลี่ย 0.743543 วินาที
  return_first_item:
    0.785511 วินาที
    0.822178 วินาที
    0.782846 วินาที
    เฉลี่ย 0.796845 วินาทีเฉลี่ย
full_lists:
  index_first_item:
    0.762618 วินาที
    0.788040 วินาที
    0.786849 วินาที
    เฉลี่ย 0.779169 วินาที
  return_first_item:
    0.802735 วินาที
    0.878706 วินาที
    0.808781 วินาที
    เฉลี่ย 0.830074 วินาที
mixed_lists:
  index_first_item:
    0.791129 วินาที
    0.743526 วินาที
    0.744441 วินาที
    เฉลี่ย 0.759699 วินาที
  return_first_item:
    0.784801 วินาที
    0.785146 วินาที
    0.840193 วินาที
    เฉลี่ย 0.803380 วินาที

try:
    return a[0]
except IndexError:
    return None

def head(iterable):
    try:
        return iter(iterable).next()
    except StopIteration:
        return None

print head(xrange(42, 1000)  # 42
print head([])               # None

BTW: ฉันจะปรับปรุงการไหลเวียนของโปรแกรมทั่วไปให้เป็นอย่างนี้:

lists = [
    ["first", "list"],
    ["second", "list"],
    ["third", "list"]
]

def do_something(element):
    if not element:
        return
    else:
        # do something
        pass

for li in lists:
    do_something(head(li))

(หลีกเลี่ยงการทำซ้ำเมื่อใดก็ตามที่เป็นไปได้)

เกี่ยวกับสิ่งนี้:

(my_list and my_list[0]) or None

หมายเหตุ:สิ่งนี้ควรใช้งานได้ดีสำหรับรายการวัตถุ แต่อาจส่งคืนคำตอบที่ไม่ถูกต้องในกรณีของรายการหมายเลขหรือสตริงตามความคิดเห็นด้านล่าง

ไม่แน่ใจว่าไพทอนเป็นแบบนี้ได้อย่างไรจนกระทั่งมีฟังก์ชั่นแรกในห้องสมุดฉันรวมสิ่งนี้ไว้ในซอร์ส:

first = lambda l, default=None: next(iter(l or []), default)

เป็นเพียงหนึ่งบรรทัด (สอดคล้องกับสีดำ) และหลีกเลี่ยงการอ้างอิง

ใช้และและหรือเคล็ดลับ:

a = get_list()
return a and a[0] or None

อาจไม่ใช่วิธีที่เร็วที่สุด แต่ไม่มีใครพูดถึงตัวเลือกนี้:

dict(enumerate(get_list())).get(0)

ถ้าget_list()สามารถกลับมาNoneคุณสามารถใช้:

dict(enumerate(get_list() or [])).get(0)

ข้อดี:

บรรทัดเดียว

- คุณแค่โทรget_list()ครั้งเดียว

- เข้าใจง่าย

กรณีการใช้งานของฉันเป็นเพียงการตั้งค่าของตัวแปรท้องถิ่น

ส่วนตัวฉันพบความพยายามและยกเว้นตัวทำความสะอาดสไตล์ให้อ่าน

items = [10, 20]
try: first_item = items[0]
except IndexError: first_item = None
print first_item

กว่าการแบ่งรายการ

items = [10, 20]
first_item = (items[:1] or [None, ])[0]
print first_item

หลายคนแนะนำให้ทำสิ่งนี้:

list = get_list()
return list and list[0] or None

ใช้งานได้ในหลายกรณี แต่จะใช้งานได้หากรายการ [0] ไม่เท่ากับ 0, เท็จหรือสตริงว่าง ถ้ารายการ [0] เป็น 0, เท็จหรือสตริงว่างวิธีจะส่งกลับไม่ถูกต้องไม่มี

ฉันได้สร้างข้อบกพร่องนี้ในรหัสของตัวเองหลายครั้งเกินไป!

คุณสามารถใช้สารสกัดจากวิธี ในคำอื่น ๆ แยกรหัสนั้นเป็นวิธีที่คุณจะโทรแล้ว

ฉันจะไม่พยายามบีบอัดมันมากนัก liners ตัวเดียวดูเหมือนอ่านยากกว่าเวอร์ชั่น verbose และถ้าคุณใช้วิธีแยกมันเป็นหนึ่งซับ;)

if mylist != []:

       print(mylist[0])

   else:

       print(None)

ไม่ได้เป็นงูหลามที่มีลักษณะคล้ายกับตัวดำเนินการที่ประกอบไปด้วย C

cond and true_expr or false_expr

กล่าวคือ

list = get_list()
return list and list[0] or None