Función para obtener números primos en python