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()