使用python将字符串中的一些单词大写

Question

actual code I wrote

所以这是我的练习，我有一个列表a = [0, 2, 4] 和一个字符串b = 'this is a sample string'，问题是根据列表a 中给出的位置将b 中的单词大写，这样输出就是“这是一个示例字符串”。我就是这样解决的：

a = [0, 2, 4]

b = 'this is a sample string'

b1 = b.split(' ')

print(b1)

#output is ['this', 'is', 'a', 'sample', 'string']

for x in a:
    b1[x] = b1[x].capitalize()

print(b)

# output is=> ['This', 'is', 'A', 'sample', 'String']

b2 = ' '.join(map(str,b))

print(b2)

# output is=> "This is A sample String"

我的问题是：这个修改可以用更小的代码来完成吗？提前致谢

Answer 1

您可以使用带有enumerate 的列表组合来检查索引是否存在于a 中。

res = ' '.join(i.capitalize() if idx in a else i for idx, i in enumerate(b.split()))

输出

This is A sample String