两个代码分别执行。。

在python中怎么实现
如果inpuu输入1执行a
输入2执行b

求指点🙏

a和b是自定义函数？
很简单啊
n = int(input())
if n==1:
a()
elif n==2:
b()