if 和 elif 函数的 Python 列表推导式

Question

from datetime import date

a_dictionary = {"name" : "John", "age" : 35, "height" : 65}

dict_items = a_dictionary.items()


def month_selct(d):
    if date.today().month == 1:
        d = list(d)[:2]

    elif date.today().month == 2:
        d = list(d)[:2]

    elif date.today().month == 3:
        d = list(d)[:2]

month_selct(dict_items)

Can you help me with reduction this code? I want this function work with each month but its just an example for 3 of it

Answer 1

You can implement a switch logic in python with a dictionary:

def month_selct(d):
    switch = {
        1 : list(d)[:2],
        2 : list(d)[:2],
        3 : list(d)[:2],
        11: list(d)[:2]
    }
    return switch[date.today().month]

print(month_selct(dict_items))

Answer 2

You can use a for loop to iterate through each month and check if that's the month you're in, as follows:

def month_select(d):
    for m in range(1, 13):
        if date.today().month == m:
            d = list(d)[:2]

EDIT: you asked for a list comprehension solution

d = [list(d)[:2] for m in range(1, 13) if date.today().month == m]