ฉันจะใช้ if / else ในพจนานุกรมเพื่อความเข้าใจได้อย่างไร


138

มีวิธีใน Python 2.7+ เพื่อสร้างสิ่งต่อไปนี้หรือไม่?

{ something_if_true if condition else something_if_false for key, value in dict_.items() }

ฉันรู้ว่าคุณสามารถทำอะไรก็ได้โดยใช้แค่ 'if':

{ something_if_true for key, value in dict_.items() if condition}

4
ตามคำบอกเล่าของ @Marcin a dictสร้างขึ้นจากkey:valueองค์ประกอบคุณไม่ได้สร้างdictที่นี่ แต่เป็นset(ดูชุดตัวอักษร )
mdeous

คำตอบ:


248

คุณมีแล้ว: A if test else Bเป็นนิพจน์ Python ที่ถูกต้อง ปัญหาเดียวเกี่ยวกับความเข้าใจในคำสั่งของคุณตามที่แสดงคือสถานที่สำหรับนิพจน์ในการเข้าใจคำสั่งต้องมีสองนิพจน์คั่นด้วยเครื่องหมายจุดคู่:

{ (some_key if condition else default_key):(something_if_true if condition
          else something_if_false) for key, value in dict_.items() }

ifประโยคสุดท้ายทำหน้าที่เป็นตัวกรองซึ่งแตกต่างจากการมีนิพจน์เงื่อนไข


29
ควรค่าแก่การกล่าวถึงว่าคุณไม่จำเป็นต้องมีเงื่อนไข if-else สำหรับทั้งคีย์และค่า ตัวอย่างเช่น{(a if condition else b): value for key, value in dict.items()}จะทำงาน
Jeremy Weirich

5
@JeremyWeirich คุณไม่จำเป็นต้องมี if-else สำหรับพวกเขาอย่างใดอย่างหนึ่งถ้าคุณไม่ต้องการ
Marcin

@Marcin เป็นไปได้ไหมที่ฉันจะใช้เฉพาะ "if" สำหรับส่วนสำคัญและใช้ทั้ง "if" และ "else" สำหรับส่วนค่า
nithin11

14

คำตอบของ @ Marcinครอบคลุมทุกอย่าง แต่ในกรณีที่มีคนต้องการดูตัวอย่างจริงฉันเพิ่มสองข้อด้านล่าง:

สมมติว่าคุณมีพจนานุกรมชุดต่อไปนี้

d = {'key1': {'a', 'b', 'c'}, 'key2': {'foo', 'bar'}, 'key3': {'so', 'sad'}}

และคุณต้องการสร้างพจนานุกรมใหม่ที่มีคีย์ระบุว่ามีสตริง'a'อยู่ในค่าหรือไม่คุณสามารถใช้

dout = {"a_in_values_of_{}".format(k) if 'a' in v else "a_not_in_values_of_{}".format(k): v for k, v in d.items()}

ซึ่งให้ผลตอบแทน

{'a_in_values_of_key1': {'a', 'b', 'c'},
 'a_not_in_values_of_key2': {'bar', 'foo'},
 'a_not_in_values_of_key3': {'sad', 'so'}}

ทีนี้สมมติว่าคุณมีพจนานุกรมสองเล่มแบบนี้

d1 = {'bad_key1': {'a', 'b', 'c'}, 'bad_key2': {'foo', 'bar'}, 'bad_key3': {'so', 'sad'}}
d2 = {'good_key1': {'foo', 'bar', 'xyz'}, 'good_key2': {'a', 'b', 'c'}}

และคุณต้องการแทนที่คีย์d1โดยคีย์d2ว่ามีค่าตามลำดับเหมือนกันหรือไม่คุณสามารถทำได้

# here we assume that the values in d2 are unique
# Python 2
dout2 = {d2.keys()[d2.values().index(v1)] if v1 in d2.values() else k1: v1 for k1, v1 in d1.items()}

# Python 3
dout2 = {list(d2.keys())[list(d2.values()).index(v1)] if v1 in d2.values() else k1: v1 for k1, v1 in d1.items()}

ซึ่งจะช่วยให้

{'bad_key2': {'bar', 'foo'},
 'bad_key3': {'sad', 'so'},
 'good_key2': {'a', 'b', 'c'}}

สำหรับตัวอย่างที่สองของคุณโดยใช้d1, d2ฉันได้รับAttributeError: 'dict_values' object has no attribute 'index'
alancalvitti

@alancalvitti: ขอบคุณที่ชี้ให้เห็น! วิธีแก้ปัญหานี้ใช้สำหรับ Python 2 และใช้ไม่ได้กับ Python 3 ฉันได้เพิ่มโซลูชัน Python 3 ด้วย
Cleb

3

ในกรณีที่คุณมีเงื่อนไขที่แตกต่างกันในการประเมินคีย์และค่าคำตอบของ @ Marcin คือหนทางที่จะไป

หากคุณมีเดียวกันเงื่อนไขสำหรับคีย์และค่าคุณดีกว่ากับการสร้าง (คีย์ค่า) -tuples ในการให้อาหารกำเนิดแสดงออกเข้าdict():

dict((modify_k(k), modify_v(v)) if condition else (k, v) for k, v in dct.items())

อ่านง่ายกว่าและระบบจะประเมินเงื่อนไขเพียงครั้งเดียวต่อคีย์ค่า

ตัวอย่างการยืมชุดพจนานุกรมของ @ Cleb:

d = {'key1': {'a', 'b', 'c'}, 'key2': {'foo', 'bar'}, 'key3': {'so', 'sad'}}

สมมติว่าคุณต้องการต่อท้ายkeysด้วยaในเท่านั้นvalueและคุณต้องการให้valueแทนที่ด้วยความยาวของชุดในกรณีเช่นนี้ มิฉะนั้นคู่คีย์ - ค่าจะไม่เปลี่ยนแปลง

dict((f"{k}_a", len(v)) if "a" in v else (k, v) for k, v in d.items())
# {'key1_a': 3, 'key2': {'bar', 'foo'}, 'key3': {'sad', 'so'}}

0

อีกตัวอย่างหนึ่งในการใช้ if / else ในพจนานุกรมเพื่อความเข้าใจ

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

รอบแรกโดยใช้การเข้ารหัสแบบดั้งเดิม (8 บรรทัด):

entries = {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}

a_dic, b_dic = {}, {}

for field, value in entries.items():
    if field == 'ther':
        for k,v in value.items():
            b_dic[k] = v
        a_dic[field] = b_dic
    else:
        a_dic[field] = value
    
print(a_dic)
 {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}”

รอบที่สองฉันพยายามใช้ความเข้าใจในพจนานุกรม แต่ยังวนอยู่ที่นั่น (6 บรรทัด):

entries = {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}

for field, value in entries.items():
    if field == 'ther':
        b_dic = {k:v for k,v in value.items()}
        a_dic[field] = b_dic
    else:
        a_dic[field] = value
    
print(a_dic)
 {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}”

สุดท้ายด้วยคำสั่งเพื่อความเข้าใจพจนานุกรมบรรทัดเดียว (1 บรรทัด):

entries = {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}

a_dic = {field:{k:v for k,v in value.items()} if field == 'ther' 
        else value for field, value in entries.items()}
    
print(a_dic)
 {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}”

ฉันใช้ python 3.8.3

โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.