python字符串操作

1.从键盘输入5个英文单词,输出其中以元音字母((包括大小写字 母)开头的单词.
程序运行显示如下:
学号: 姓名:
请输入第1个英文单词:CMU
请输入第2个英文单词:One
请输入第3个英文单词:python
LICMU
首字母是元音的单词有:
ONE
egg


lst = []
for i in range(5):
    str1 = input('请输入第{}个英文单词:'.format(i+1))
    if str1[0].upper() in 'AEIOU':
        lst.append(str1)
print('首字母是元音的单词有:\n{}'.format('\n'.join(lst)))