Python-测验 | Python字符串测验 | 问题2

本篇测验是关于Python字符串的问题。在这个问题中，我们会考察对Python字符串的操作和使用。

问题描述

给定一个字符串s，编写一个Python函数，将字符串s中的每个单词的首字母大写，其他字母小写，并返回新的字符串。

输入格式

• 字符串s，长度不超过100。

输出格式

• 首字母大写，其他字母小写的新字符串。

示例

输入：

s = "this is a test"

输出：

"This Is A Test"

解题思路

我们可以分步来解决这个问题：

1. 将字符串s按空格分割成单词列表。
2. 遍历每个单词，将其转换成首字母大写，其他字母小写。
3. 将单词列表拼接为一个新的字符串，每个单词之间用空格隔开。

下面是示例代码：

def capitalize_words(s):
    # 将字符串按空格分割成单词列表
    words = s.split()
    # 遍历每个单词，将其转换成首字母大写，其他字母小写
    capped_words = [word.capitalize() for word in words]
    # 将单词列表拼接为一个新的字符串，每个单词之间用空格隔开
    return ' '.join(capped_words)

测试代码

下面是一个简单的测试代码，用于验证函数的正确性：

s = "this is a test"
expected_output = "This Is A Test"
assert capitalize_words(s) == expected_output

如果测试通过，说明解题思路和代码实现是正确的。