Are all elements prime?

Write a Python program to check whether all elements of a set are prime, and list the primes.

{2, 3, 5, 9}

False [2, 3, 5]

Approach

1. A prime test for each element.

Solution

def is_prime(n):    return n > 1 and all(n % d for d in range(2, int(n ** 0.5) + 1)) s = {2, 3, 5, 9}print(all(is_prime(x) for x in s), sorted(x for x in s if is_prime(x)))

False [2, 3, 5]

More set programs

← All 48 Set Programs