วิธี Pythonic เพื่อหลีกเลี่ยงคำสั่ง“ if x: return x”

ฉันมีวิธีการที่เรียก 4 วิธีอื่น ๆ ตามลำดับเพื่อตรวจสอบเงื่อนไขเฉพาะและคืนค่าทันที

def check_all_conditions():
    x = check_size()
    if x:
        return x

    x = check_color()
    if x:
        return x

    x = check_tone()
    if x:
        return x

    x = check_flavor()
    if x:
        return x
    return None

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

x and return x

แต่นั่นเป็น Python ที่ไม่ถูกต้อง ฉันคิดถึงวิธีแก้ปัญหาที่เรียบง่ายและสง่างามใช่หรือไม่? บังเอิญในสถานการณ์นี้วิธีการตรวจสอบทั้งสี่นั้นอาจมีราคาแพงดังนั้นฉันจึงไม่ต้องการเรียกพวกเขาหลายครั้ง

คุณสามารถใช้วง:

conditions = (check_size, check_color, check_tone, check_flavor)
for condition in conditions:
    result = condition()
    if result:
        return result

นี่เป็นข้อดีที่เพิ่มขึ้นซึ่งตอนนี้คุณสามารถกำหนดจำนวนเงื่อนไขได้แล้ว

คุณสามารถใช้map()+ filter()(เวอร์ชัน Python 3 ใช้future_builtinsเวอร์ชันใน Python 2) เพื่อรับค่าการจับคู่เช่น:

try:
    # Python 2
    from future_builtins import map, filter
except ImportError:
    # Python 3
    pass

conditions = (check_size, check_color, check_tone, check_flavor)
return next(filter(None, map(lambda f: f(), conditions)), None)

แต่ถ้าเป็นสิ่งที่อ่านได้มากกว่าจะเป็นที่ถกเถียงกัน

ตัวเลือกอื่นคือการใช้นิพจน์ตัวสร้าง:

conditions = (check_size, check_color, check_tone, check_flavor)
checks = (condition() for condition in conditions)
return next((check for check in checks if check), None)

อีกทางเลือกหนึ่งคือคำตอบที่ดีของ Martijn orของคุณสามารถห่วงโซ่ นี่จะคืนค่าความจริงแรกหรือNoneหากไม่มีค่าจริง:

def check_all_conditions():
    return check_size() or check_color() or check_tone() or check_flavor() or None

การสาธิต:

>>> x = [] or 0 or {} or -1 or None
>>> x
-1
>>> x = [] or 0 or {} or '' or None
>>> x is None
True

อย่าเปลี่ยนมัน

มีวิธีอื่นในการทำเช่นนี้ตามคำตอบอื่น ๆ ที่แสดง ไม่มีความชัดเจนเท่ากับรหัสต้นฉบับของคุณ

ตอบได้อย่างมีประสิทธิภาพเช่นเดียวกับ timgeb แต่คุณสามารถใช้วงเล็บสำหรับการจัดรูปแบบ nicer:

def check_all_the_things():
    return (
        one()
        or two()
        or five()
        or three()
        or None
    )

ตามกฎของ Curlyคุณสามารถทำให้โค้ดนี้อ่านง่ายขึ้นโดยแยกข้อกังวลสองข้อ:

• ฉันตรวจสอบอะไรบ้าง
• มีสิ่งหนึ่งคืนจริงหรือไม่?

เป็นสองฟังก์ชั่น:

def all_conditions():
    yield check_size()
    yield check_color()
    yield check_tone()
    yield check_flavor()

def check_all_conditions():
    for condition in all_conditions():
        if condition:
            return condition
    return None

หลีกเลี่ยงสิ่งนี้:

• โครงสร้างเชิงตรรกะที่ซับซ้อน
• เส้นยาวจริงๆ
• การทำซ้ำ

... ในขณะที่รักษาลำดับการไหลให้อ่านง่าย

คุณอาจจะได้ชื่อฟังก์ชั่นที่ดีขึ้นตามสถานการณ์เฉพาะของคุณซึ่งทำให้อ่านง่ายขึ้น

นี่คือตัวแปรของตัวอย่างแรกของ Martijns นอกจากนี้ยังใช้ "คอลเลกชัน callables" - สไตล์เพื่อให้ลัดวงจร

แทนที่จะห่วงคุณสามารถใช้ anybuiltin

conditions = (check_size, check_color, check_tone, check_flavor)
return any(condition() for condition in conditions) 

โปรดทราบว่าanyส่งคืนบูลีนดังนั้นหากคุณต้องการค่าส่งคืนที่แน่นอนของการตรวจสอบการแก้ปัญหานี้จะไม่ทำงาน anyจะไม่เห็นความแตกต่างระหว่าง14, 'red', 'sharp', เป็นค่าตอบแทนที่พวกเขาจะทั้งหมดจะกลับมาเป็น'spicy'True

คุณคิดว่าจะเขียนif x: return xทั้งหมดในบรรทัดเดียวหรือไม่?

def check_all_conditions():
    x = check_size()
    if x: return x

    x = check_color()
    if x: return x

    x = check_tone()
    if x: return x

    x = check_flavor()
    if x: return x

    return None

นี่ไม่ได้น้อยไปกว่ากัน ซ้ำซ้อนกว่าสิ่งที่คุณมี แต่ IMNSHO มันอ่านได้ค่อนข้างราบรื่นขึ้น

ฉันค่อนข้างประหลาดใจที่ไม่มีใครพูดถึงตัวในตัวanyซึ่งทำขึ้นเพื่อจุดประสงค์นี้:

def check_all_conditions():
    return any([
        check_size(),
        check_color(),
        check_tone(),
        check_flavor()
    ])

โปรดทราบว่าแม้ว่าการใช้งานนี้อาจเป็นสิ่งที่ชัดเจนที่สุด แต่ก็จะประเมินการตรวจสอบทั้งหมดแม้ว่าจะมีการตรวจสอบครั้งแรกTrueก็ตาม

หากคุณต้องการหยุดที่การตรวจสอบครั้งแรกล้มเหลวจริงๆให้ลองใช้วิธีการreduceที่ทำเพื่อแปลงรายการให้เป็นค่าอย่างง่าย:

def check_all_conditions():
    checks = [check_size, check_color, check_tone, check_flavor]
    return reduce(lambda a, f: a or f(), checks, False)

reduce(function, iterable[, initializer]): ใช้ฟังก์ชั่นของทั้งสองข้อโต้แย้งรวมกับรายการที่ทำซ้ำได้จากซ้ายไปขวาเพื่อลดการทำซ้ำเป็นค่าเดียว อาร์กิวเมนต์ซ้าย, x, เป็นค่าสะสมและอาร์กิวเมนต์ที่ถูกต้องคือค่าการอัพเดทจาก iterable ถ้ามี initializer ที่เป็นทางเลือกมันจะถูกวางไว้ก่อนรายการของ iterable ในการคำนวณ

ในกรณีของคุณ:

• lambda a, f: a or f()เป็นหน้าที่ที่จะตรวจสอบว่าทั้งสะสมaหรือการตรวจสอบในปัจจุบันคือf() Trueโปรดทราบว่าถ้าaเป็นTrue, f()จะไม่ได้รับการประเมิน
• checksมีฟังก์ชั่นตรวจสอบ ( fรายการจากแลมบ์ดา)
• False เป็นค่าเริ่มต้นมิฉะนั้นจะไม่มีการตรวจสอบเกิดขึ้นและผลลัพธ์จะเป็นเสมอ True

anyและreduceเป็นเครื่องมือพื้นฐานสำหรับการเขียนโปรแกรมการทำงาน ฉันขอแนะนำให้คุณฝึกฝนสิ่งเหล่านี้รวมถึงสิ่งที่ยอดเยี่ยมเช่นmapกัน!

หากคุณต้องการโครงสร้างของรหัสเดียวกันคุณสามารถใช้ข้อความประกอบไปด้วย!

def check_all_conditions():
    x = check_size()
    x = x if x else check_color()
    x = x if x else check_tone()
    x = x if x else check_flavor()

    return x if x else None

ฉันคิดว่านี่ดูดีและชัดเจนถ้าคุณดู

การสาธิต:

สำหรับฉันคำตอบที่ดีที่สุดคือจาก @ phil-frost ตามด้วย @ wayne-werner's

สิ่งที่ฉันคิดว่าน่าสนใจคือไม่มีใครพูดอะไรเกี่ยวกับความจริงที่ว่าฟังก์ชั่นจะคืนค่าข้อมูลหลายประเภทซึ่งจะทำให้ต้องตรวจสอบชนิดของตัว x เพื่อทำงานต่อไป

ดังนั้นฉันจะผสม @ PhilFrost การตอบสนองกับความคิดในการรักษาประเภทเดียว:

def all_conditions(x):
    yield check_size(x)
    yield check_color(x)
    yield check_tone(x)
    yield check_flavor(x)

def assessed_x(x,func=all_conditions):
    for condition in func(x):
        if condition:
            return x
    return None

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

ด้วยวิธีนี้คุณจะได้รับการxตรวจสอบเพียงครั้งเดียว แต่จะเป็นประเภทเดียวกันเสมอ

เป็นการดีที่ฉันจะเขียนcheck_ ฟังก์ชั่นอีกครั้งเพื่อกลับTrueหรือFalseมากกว่าค่า เช็คของคุณจะกลายเป็น

if check_size(x):
    return x
#etc

สมมติว่าคุณxไม่ได้เปลี่ยนรูป, ฟังก์ชั่นของคุณยังสามารถปรับเปลี่ยนได้ (แม้ว่าพวกเขาไม่สามารถเปลี่ยนมัน) - แต่ฟังก์ชั่นที่เรียกว่าcheckไม่ควรจริงๆจะปรับเปลี่ยนมันอยู่แล้ว

การเปลี่ยนแปลงเล็กน้อยในตัวอย่างแรกของ Martijns ด้านบนซึ่งจะช่วยหลีกเลี่ยง if ในลูป:

Status = None
for c in [check_size, check_color, check_tone, check_flavor]:
  Status = Status or c();
return Status

ฉันชอบ @ timgeb ในระหว่างนี้ฉันต้องการเพิ่มว่าการแสดงNoneในreturnคำสั่งนั้นไม่จำเป็นเนื่องจากการรวบรวมorคำสั่งที่แยกกันจะได้รับการประเมินและไม่มีศูนย์แรก, ไม่มีว่าง, ไม่มีไม่มีส่งคืนและถ้าไม่มีจะNoneถูกส่งกลับ ไม่ว่าจะมีNoneหรือไม่!

ดังนั้นcheck_all_conditions()ฟังก์ชั่นของฉันจะเป็นดังนี้:

def check_all_conditions():
    return check_size() or check_color() or check_tone() or check_flavor()

ใช้timeitกับnumber=10**7ฉันดูเวลาทำงานของจำนวนข้อเสนอแนะ เพื่อประโยชน์ในการเปรียบเทียบฉันใช้random.random()ฟังก์ชันเพื่อส่งคืนสตริงหรือNoneตามตัวเลขสุ่ม นี่คือรหัสทั้งหมด:

import random
import timeit

def check_size():
    if random.random() < 0.25: return "BIG"

def check_color():
    if random.random() < 0.25: return "RED"

def check_tone():
    if random.random() < 0.25: return "SOFT"

def check_flavor():
    if random.random() < 0.25: return "SWEET"

def check_all_conditions_Bernard():
    x = check_size()
    if x:
        return x

    x = check_color()
    if x:
        return x

    x = check_tone()
    if x:
        return x

    x = check_flavor()
    if x:
        return x
    return None

def check_all_Martijn_Pieters():
    conditions = (check_size, check_color, check_tone, check_flavor)
    for condition in conditions:
        result = condition()
        if result:
            return result

def check_all_conditions_timgeb():
    return check_size() or check_color() or check_tone() or check_flavor() or None

def check_all_conditions_Reza():
    return check_size() or check_color() or check_tone() or check_flavor()

def check_all_conditions_Phinet():
    x = check_size()
    x = x if x else check_color()
    x = x if x else check_tone()
    x = x if x else check_flavor()

    return x if x else None

def all_conditions():
    yield check_size()
    yield check_color()
    yield check_tone()
    yield check_flavor()

def check_all_conditions_Phil_Frost():
    for condition in all_conditions():
        if condition:
            return condition

def main():
    num = 10000000
    random.seed(20)
    print("Bernard:", timeit.timeit('check_all_conditions_Bernard()', 'from __main__ import check_all_conditions_Bernard', number=num))
    random.seed(20)
    print("Martijn Pieters:", timeit.timeit('check_all_Martijn_Pieters()', 'from __main__ import check_all_Martijn_Pieters', number=num))
    random.seed(20)
    print("timgeb:", timeit.timeit('check_all_conditions_timgeb()', 'from __main__ import check_all_conditions_timgeb', number=num))
    random.seed(20)
    print("Reza:", timeit.timeit('check_all_conditions_Reza()', 'from __main__ import check_all_conditions_Reza', number=num))
    random.seed(20)
    print("Phinet:", timeit.timeit('check_all_conditions_Phinet()', 'from __main__ import check_all_conditions_Phinet', number=num))
    random.seed(20)
    print("Phil Frost:", timeit.timeit('check_all_conditions_Phil_Frost()', 'from __main__ import check_all_conditions_Phil_Frost', number=num))

if __name__ == '__main__':
    main()

และนี่คือผลลัพธ์:

Bernard: 7.398444877040768
Martijn Pieters: 8.506569201346597
timgeb: 7.244275416364456
Reza: 6.982133448743038
Phinet: 7.925932800076634
Phil Frost: 11.924794811353031

วิธีนี้อยู่นอกกรอบเล็กน้อย แต่ฉันคิดว่าผลลัพธ์สุดท้ายนั้นง่ายอ่านง่ายและดูดี

แนวคิดพื้นฐานคือraiseข้อยกเว้นเมื่อฟังก์ชันใดฟังก์ชันหนึ่งประเมินว่าเป็นความจริงและส่งคืนผลลัพธ์ นี่คือลักษณะ:

def check_conditions():
    try:
        assertFalsey(
            check_size,
            check_color,
            check_tone,
            check_flavor)
    except TruthyException as e:
        return e.trigger
    else:
        return None

คุณจะต้องใช้assertFalseyฟังก์ชันที่ยกข้อยกเว้นเมื่อหนึ่งในอาร์กิวเมนต์ของฟังก์ชันที่เรียกว่าประเมินว่าเป็นความจริง:

def assertFalsey(*funcs):
    for f in funcs:
        o = f()
        if o:
            raise TruthyException(o)

ด้านบนสามารถแก้ไขได้เพื่อให้มีข้อโต้แย้งสำหรับฟังก์ชั่นที่จะได้รับการประเมิน

และแน่นอนคุณจะต้องการTruthyExceptionตัวของมันเอง ข้อยกเว้นนี้ให้สิ่งobjectที่เรียกใช้ข้อยกเว้น:

class TruthyException(Exception):
    def __init__(self, obj, *args):
        super().__init__(*args)
        self.trigger = obj

คุณสามารถเปลี่ยนฟังก์ชั่นดั้งเดิมให้กลายเป็นสิ่งทั่วไปได้มากกว่า:

def get_truthy_condition(*conditions):
    try:
        assertFalsey(*conditions)
    except TruthyException as e:
        return e.trigger
    else:
        return None

result = get_truthy_condition(check_size, check_color, check_tone, check_flavor)

สิ่งนี้อาจช้าลงเล็กน้อยเนื่องจากคุณใช้ทั้งifคำสั่งและการจัดการข้อยกเว้น อย่างไรก็ตามข้อยกเว้นจะได้รับการจัดการสูงสุดเพียงครั้งเดียวดังนั้นการเข้าชมเพื่อประสิทธิภาพควรเป็นเรื่องเล็กน้อยเว้นแต่คุณคาดว่าจะเรียกใช้การตรวจสอบและรับTrueค่าหลายพันครั้ง

วิธีไพทอนอาจใช้การลดขนาด (ตามที่มีคนพูดถึงแล้ว) หรือ itertools (ดังที่แสดงด้านล่าง) แต่สำหรับผมแล้วดูเหมือนว่าเพียงแค่ใช้การลัดวงจรของorผู้ปฏิบัติงานสร้างรหัสที่ชัดเจนขึ้น

from itertools import imap, dropwhile

def check_all_conditions():
    conditions = (check_size,\
        check_color,\
        check_tone,\
        check_flavor)
    results_gen = dropwhile(lambda x:not x, imap(lambda check:check(), conditions))
    try:
        return results_gen.next()
    except StopIteration:
        return None

ฉันจะกระโดดเข้าไปที่นี่และไม่เคยเขียน Python สักบรรทัดเดียว แต่ฉันถือว่า if x = check_something(): return xถูกต้องหรือไม่

ถ้าเป็นเช่นนั้น:

def check_all_conditions():

    if (x := check_size()): return x
    if (x := check_color()): return x
    if (x := check_tone()): return x
    if (x := check_flavor()): return x

    return None

หรือใช้max:

def check_all_conditions():
    return max(check_size(), check_color(), check_tone(), check_flavor()) or None

ฉันเคยเห็นการใช้งานที่น่าสนใจของงบสวิตช์ / กรณีที่มี dicts ในอดีตที่ทำให้ฉันคำตอบนี้ จากตัวอย่างที่คุณให้มาคุณจะได้รับสิ่งต่อไปนี้ (มันเป็นบ้าusing_complete_sentences_for_function_namesจึงcheck_all_conditionsเปลี่ยนชื่อเป็นstatusดู (1))

def status(k = 'a', s = {'a':'b','b':'c','c':'d','d':None}) :
  select = lambda next, test : test if test else next
  d = {'a': lambda : select(s['a'], check_size()  ),
       'b': lambda : select(s['b'], check_color() ),
       'c': lambda : select(s['c'], check_tone()  ),
       'd': lambda : select(s['d'], check_flavor())}
  while k in d : k = d[k]()
  return k

ฟังก์ชั่นที่เลือกช่วยลดความจำเป็นที่จะต้องโทรหากันcheck_FUNCTIONสองครั้งนั่นคือคุณหลีกเลี่ยงcheck_FUNCTION() if check_FUNCTION() else nextโดยการเพิ่มฟังก์ชั่นเลเยอร์อื่น สิ่งนี้มีประโยชน์สำหรับการใช้งานที่ยาวนาน lambdas ในการดำเนินการ dict delay ของค่าจนกว่าลูป while

เป็นโบนัสที่คุณสามารถแก้ไขคำสั่งการดำเนินการและแม้แต่ข้ามการทดสอบบางอย่างโดยการเปลี่ยนแปลงkและsเช่นk='c',s={'c':'b','b':None}ลดจำนวนการทดสอบและกลับคำสั่งการประมวลผลดั้งเดิม

timeitทุนอาจต่อรองสูงกว่าต้นทุนของการเพิ่มชั้นพิเศษหรือสองไปยังกองและค่าใช้จ่ายสำหรับ Dict ที่มองขึ้น แต่คุณดูเหมือนกังวลมากขึ้นกับความน่ารักของรหัส

อีกวิธีหนึ่งที่ใช้งานง่ายกว่าอาจเป็นดังนี้:

def status(k=check_size) :
  select = lambda next, test : test if test else next
  d = {check_size  : lambda : select(check_color,  check_size()  ),
       check_color : lambda : select(check_tone,   check_color() ),
       check_tone  : lambda : select(check_flavor, check_tone()  ),
       check_flavor: lambda : select(None,         check_flavor())}
  while k in d : k = d[k]()
  return k

1. ฉันหมายถึงสิ่งนี้ไม่ได้อยู่ในรูปของ pep8 แต่ในแง่ของการใช้คำอธิบายสั้น ๆ อย่างเดียวแทนประโยค ที่ได้รับ OP อาจปฏิบัติตามอนุสัญญาการเข้ารหัสทำงานหนึ่งฐานรหัสที่มีอยู่บางส่วนหรือไม่สนใจคำศัพท์สั้น ๆ ใน codebase ของพวกเขา