python中类似开关大小写的结构[重复]

Question

我使用了下面的函数，它获取成员的登录日期和合同长度，并返回一个修改后的数据框，其中包含每个成员的到期日期。该函数按预期工作，但我想知道是否有更干净的方法来编写嵌套函数month\u to_num（ind）。我知道python没有实现cases，但是有没有办法重写所有if/elif/else语句？

def renewal_date(df):

    def month_to_num(ind):
        if (df.loc[ind, "Contract Type"] == "1 Month"):
            return 1
        elif (df.loc[ind, "Contract Type"] == "3 Months Prom" or 
              df.loc[ind, "Contract Type"] == "3 Month"):
            return 3
        elif (df.loc[ind, "Contract Type"] == "4 Month Promo"):
            return 4
        elif (df.loc[ind, "Contract Type"] == "6 Months"):
            return 6
        else:
            return 12

    for z in range(0, len(df)): 
        exp_date = (df.loc[z, "Date-Joined"] + 
                    relativedelta(months=+month_to_num(z)))
        df.set_value(z,"Date-Renewal", exp_date)

    return df

Answer 1

您可以将字典与dict.get（）方法配合使用，如果字典中不存在键，该方法将返回默认值。

def renewal_date(df):
    def month_to_num(ind):
        contract_types = {'1 Month': 1, '3 Months Prom': 3, '4 Month Promo': 4, '6 Months': 6}

        return contract_types.get(df.loc[ind, "Contract Type"], 12)

    for z in range(0, len(df)):
        exp_date = (df.loc[z, "Date-Joined"] +
                    relativedelta(months=+month_to_num(z)))
        df.set_value(z, "Date-Renewal", exp_date)

    return df

Answer 2

这更适合于代码复查堆栈交换，但为了保留功能，您通常会使用字典：

MONTH_NUMS = {
    '1 Month': 1,
    '3 Months Prom': 3,
    '3 Month': 3,
    '4 Month Promo': 4,
    '6 Months': 6,
}

def month_to_num(ind):
    return MONTH_NUMS.get(df.loc[ind, 'Contract Type'], 12)

也就是说，在我们如何以这种方式将字符串映射到数字的过程中，有一点代码的味道。