Python 函数：避免使用 if 子句进行参数检查

Question

这是我的功能：def save_to_mongo(self, df, collection, additional_variable):    for index, row in df.iterrows():        result = row.to_dict()        collection.update_one(            {"_id": str(row['date']) + "_" + str(row['sector']) + "_" + str(row['caseid']) + str(row[additional_variable])},            {                '$set': result            },            upsert=True)我有许多类似的函数，其中additonal_variable可以包含None.我真的很想避免用这样的风格来膨胀代码库：if additional_varibale is None:    collection.update_one(        {"_id": str(row['date']) + "_" + str(row['sector']) + "_" + str(row['caseid'])},        {            '$set': result        },        upsert=True)else:    collection.update_one(        {"_id": str(row['date']) + "_" + str(row['sector']) + "_" + str(row['caseid']) + str(row[additional_variable])},        {            '$set': result        },        upsert=True)我认为这段代码很难看并且难以维护。有没有更好的方法或最佳实践来避免使用这些长的ifandelse语句？

Answer 1

您可以将块最小if else化为 -

additional_varibale = '' if additional_varibale is None else str(row[additional_variable])

    

collection.update_one(

    {"_id": str(row['date']) + "_" + str(row['sector']) + "_" + str(row['caseid']) + additional_varibale},

    {

        '$set': result

    },

    upsert=True)