在一行内连续输入若干个正整数，用逗号分隔。将它们中的素数按顺序存入一个列表中并输出该列表。

Question

题目内容：
在一行内连续输入若干个正整数，用逗号分隔。将它们中的素数按顺序存入一个列表中并输出该列表。

输入格式:
一行用逗号分隔的正整数

输出格式：
素数组成的列表

输入样例：
3, 4, 5, 6, 7, 8

输出样例：
[3, 5, 7]

Answer 1

def prime(n):
  if n < 2:
    return False
  for i in range(2, n - 1):
    if n % 2 == 0:
      return False
  else:
    return True

s = input()
lst = s.split(',')
lst = filter(lambda x: prime(int(x)), lst)
print(lst)