问题描述:
Write a function to find all words starting with 'a' or 'e' in a given string.

测试用例:
['assert words_ae("python programe")==[\'ame\']', 'assert words_ae("python programe language")==[\'ame\',\'anguage\']', 'assert words_ae("assert statement")==[\'assert\', \'atement\']']

生成的代码:
def words_ae(input_string):
    words = input_string.split()
    result = [word[1:] for word in words if word.startswith('a') or word.startswith('e')]
    return result