编写一个Python程序来计算字符串中元音字母的数量。

def vowel_count(string):
    #创建列表lst，储存数据
    lst=['a','e','i','o','u'] 
    #计算累加和
    s=0 
    #条件判断如果strin在列表lst中......    
    for i in string:
        if i in lst:
            s+=1
    return s

# 获取输入字符串 
input_string = input()
# 调用函数 
print(vowel_count(input_string))