Python字符替换

怎么把用户输入的字符中的元音字母替换成大写字母

s = input()
r = ''
for t in s:
    if t.lower() in ['a', 'e', 'i', 'o', 'u']:
        r += t.upper()
    else:
        r += t
print(r)