Python_列表中的偶数变成他的平方

#编写程序将列表中的偶数变成他的平方
def word_len(s):
    # s = [i ** 2 for i in s if i % 2 == 0]
    L = []
    for i in s:
        if(i % 2 == 0):
            L.append(i** 2)
        else:
            L.append(i)
    return L
def main():
    s = [2, 6, 1, 3, 9, 4, 22]
    print(word_len(s))

main()