大家好,我是链上杯子(掘金:链上杯子)。
失业一年了,天天想着怎么翻身。最近在写提示词的时候发现,用户需求千差万别:有时候想要温柔治愈的输出,有时候又要冷硬毒舌的风格。以前每次都手动改代码,现在用 if 判断一下用户输入,就能自动切换模板,感觉像给 AI 加了个“自动档”。Python 里这个条件判断用起来其实很简单,但一结合 AI 场景,省下来的手动调整时间可不少。
本课目标:掌握 if / elif / else 条件判断,根据用户输入(比如风格偏好、内容类型、长度要求)自动选择或调整提示词模板,实现更智能的动态生成。
核心代码实战
1. 简单 if 判断 —— 根据字数选择不同约束
length = int(input("你希望生成的内容字数大约是多少?(例如 200/500/1000):"))
if length <= 200:
constraint = "字数严格控制在200字以内,语言极简有力"
elif length <= 500:
constraint = "字数控制在500字左右,内容完整但不冗长"
else:
constraint = "字数约1000字,详细展开,包含多个段落和例子"
prompt = "你是一个专业的内容创作者。" + \\
"请写一篇关于「AI 如何改变日常写作」的文章。" + \\
constraint + "。"
print("\\n生成的提示词:")
print(prompt)
我运行后,输入 300,看到的输出是:
你希望生成的内容字数大约是多少?(例如 200/500/1000):300
生成的提示词:
你是一个专业的内容创作者。请写一篇关于「AI 如何改变日常写作」的文章。字数控制在500字左右,内容完整但不冗长。
2. 多条件 elif —— 根据风格选择不同角色设定
style = input("你想要的输出风格是?(温柔/硬核/幽默/专业):").strip().lower()
if style == "温柔":
role = "你是一个温暖治愈的生活导师"
tone = "语气温柔、体贴,像在和朋友聊天"
elif style == "硬核":
role = "你是一个直言不讳的技术硬核博主"
tone = "语气冷峻、直接,数据说话,不讲情面"
elif style == "幽默":
role = "你是一个段子手级别的搞笑创作者"
tone = "全程幽默风趣,适当自黑和玩梗"
else:
role = "你是一个专业严谨的编辑"
tone = "语气客观、专业,使用正式语言"
prompt = f"{role}。\\n请围绕「如何用 Python 快速生成高质量 Prompt」写一篇指南。\\n{tone}。"
print("\\n生成的提示词:")
print(prompt)
我试了输入“幽默”,输出是:
你想要的输出风格是?(温柔/硬核/幽默/专业):幽默
生成的提示词:
你是一个段子手级别的搞笑创作者。
请围绕「如何用 Python 快速生成高质量 Prompt」写一篇指南。
全程幽默风趣,适当自黑和玩梗。
3. 组合使用 —— 风格 + 长度 双重判断
theme = input("主题是?(例如:未来城市、失恋日记):")
mood = input("情绪基调(积极/治愈/黑暗/搞笑):").strip().lower()
if mood in ["积极", "治愈"]:
prefix = "故事要温暖向上,充满希望"
elif mood == "黑暗":
prefix = "氛围阴郁、压抑,带点哲学思辨"
elif mood == "搞笑":
prefix = "全程沙雕搞笑,适当夸张和反转"
else:
prefix = "保持中性,真实自然"
length_input = int(input("大约字数(200/500/800):"))
if length_input <= 300:
length_str = "短篇,控制在300字内"
elif length_input <= 600:
length_str = "中篇,约500-600字"
else:
length_str = "长篇,800字以上,细节丰富"
prompt = f"你是一个小说创作者。\\n请写一个关于「{theme}」的原创故事。\\n{prefix}。\\n{length_str}。"
print("\\n生成的提示词:")
print(prompt)
运行几次不同输入,提示词就会根据选择自动调整。
小练习(2 道)
练习1(基础)
写一段代码,让用户输入一个数字(代表想要的创意数量,例如 3/5/10)。
用 if/elif/else 判断:
- 如果 <=3,提示词要求“精炼,每条不超过30字”
- 如果 <=7,中等长度
- 否则,详细展开 然后拼接成一个“脑暴 N 个产品 slogan”的提示词并打印。
练习2(进阶)
让用户输入内容类型(例如:朋友圈文案/小红书笔记/邮件主题)和情绪(开心/emo/励志/吐槽)。
用 if/elif/else 根据情绪选择不同的语气描述,再拼接成完整提示词。建议把生成的提示词复制到大模型试试实际输出差异。
本课小结
本课介绍了 if / elif / else 条件判断,以及如何结合 input() 实现分支逻辑。
在 AI 提示词场景中,条件判断能让程序“聪明”地根据用户需求自动切换角色、语气、长度等参数,生成的模板更贴合实际使用。
下节预告
下一课:循环(while/for)—— 多轮模拟对话、批量生成提示词变体。
欢迎在评论区贴出练习代码或生成的提示词,有问题可以留言。