在熊猫的数据帧中使用If语句

Question

我的程序必须执行以下两项操作之一。如果用户输入幸运，它应该从我的数据库中输出一个randam示例，而不做其他任何事情（我想“退出”程序。如果他们输入“自定义”，则应该从我的数据中输出一个流派列表。

我尝试使用sys.exit，退出，退出等。当我使用sys.exit时，我得到了想要的输出，但它不在数据框中...

start_search = input("Enter Lucky or Custom")

if start_search == "lucky":
     sys.exit(lucky_output(start_search)
elif start_search == "custom":
     genre_db 
else:
     print("Invalid Entry")

我也尝试将函数放在sys.exit之前，但没有输出任何内容。请帮忙！