《2026 Python零基础入门:用AI主题学编程》第四课:条件判断(if/elif/else)——根据输入选择不同的提示词模板或风格

7 阅读4分钟

大家好,我是链上杯子(掘金:链上杯子)。

失业一年了,天天想着怎么翻身。最近在写提示词的时候发现,用户需求千差万别:有时候想要温柔治愈的输出,有时候又要冷硬毒舌的风格。以前每次都手动改代码,现在用 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)—— 多轮模拟对话、批量生成提示词变体。

欢迎在评论区贴出练习代码或生成的提示词,有问题可以留言。