Python如何通过if...else条件判断选择不同输入数据集

你直接按照布尔分支的基础逻辑写就行，步骤非常清晰：

• 先定义用来控制分支的布尔类型开关变量，值为True或者False
• 写if判断分支：当开关变量值为真时，执行加载A数据集的逻辑
• 写else兜底分支：当开关变量值为假时，执行加载B数据集的逻辑

最容易理解的基础写法如下，你可以直接替换里面加载数据集的实际业务代码：

# 分支开关，可根据实际业务场景修改这个值
select_a = True

if select_a:
    # 替换成你加载A Dataset的实际代码，比如读文件、调用数据集加载接口等
    input_data = "A Dataset"
    print("已加载A数据集作为输入")
else:
    # 替换成你加载B Dataset的实际代码
    input_data = "B Dataset"
    print("已加载B数据集作为输入")

# 后续所有程序逻辑直接调用input_data变量即可，不需要重复判断

初学者注意：Python中判断布尔值不需要额外写select_a == True，直接把变量放在if后面就是最规范的写法，不会出现类型判断的冗余问题。

等你熟悉基础逻辑之后，也可以用Python的三元表达式把逻辑压缩成一行，效果和上面的多行判断完全一致：

select_a = True
input_data = "A Dataset" if select_a else "B Dataset"

内容的提问来源于stack exchange，提问作者Python Dev