dada una cadena numérica imprimir los dígitos primos

def prime_number(number):
  prime_list = []
  for n in number:
    if n in [1, 2, 3, 5, 7]:
      prime_list.append(n)
  return prime_list

prime_number("31357")