Python Program to Illustrate Different Set Operations

Spread the love

Sets are an integral part of Python, offering a wide range of functionalities for unordered collection data types. The concept is borrowed from mathematics, and Python’s set type supports methods like union, intersection, and difference, among others. This article aims to provide an exhaustive guide on different set operations in Python, with examples to illustrate each operation.

Table of Content

  1. Creating a Set
  2. Adding and Removing Elements
  3. Basic Set Operations
    • Union
    • Intersection
    • Difference
    • Symmetric Difference
  4. Other Operations
    • Subset
    • Superset
    • Disjoint Sets
  5. Advanced Set Operations
    • Using Set Comprehensions
    • Frozen Sets
  6. Real-world Applications
  7. Conclusion

Creating a Set

In Python, sets are created using curly braces {} or the built-in set() function. However, an empty set can only be created using set(). Here’s how to create sets:

# Using curly braces
my_set = {1, 2, 3, 4}
print(my_set)  # Output: {1, 2, 3, 4}

# Using set() function
another_set = set([1, 2, 3, 4])
print(another_set)  # Output: {1, 2, 3, 4}

# Empty set
empty_set = set()
print(empty_set)  # Output: set()

Adding and Removing Elements

Adding Elements

  • add(element): Adds a single element to the set.
  • update(iterable): Adds multiple elements to the set.
my_set = {1, 2, 3}
my_set.add(4)  # Output: {1, 2, 3, 4}
my_set.update([5, 6, 7])  # Output: {1, 2, 3, 4, 5, 6, 7}

Removing Elements

  • remove(element): Removes an element from the set. Raises a KeyError if the element is not found.
  • discard(element): Removes an element if it exists, does nothing otherwise.
  • pop(): Removes and returns an arbitrary element. Raises a KeyError if the set is empty.
my_set = {1, 2, 3, 4, 5}
my_set.remove(1)  # Output: {2, 3, 4, 5}
my_set.discard(2)  # Output: {3, 4, 5}
my_set.pop()  # Output: Removes and returns an arbitrary element

Basic Set Operations

Union

Union combines the elements from both sets.

A = {1, 2, 3}
B = {3, 4, 5}
union_set = A.union(B)  # Output: {1, 2, 3, 4, 5}

You can also use the | operator.

union_set = A | B  # Output: {1, 2, 3, 4, 5}

Intersection

Returns the elements that are common in both sets.

intersection_set = A.intersection(B)  # Output: {3}

Or you can use the & operator.

intersection_set = A & B  # Output: {3}

Difference

Returns the elements that are in the first set but not in the second.

difference_set = A.difference(B)  # Output: {1, 2}

Or you can use the - operator.

difference_set = A - B  # Output: {1, 2}

Symmetric Difference

Returns the elements that are unique to each set.

symmetric_difference_set = A.symmetric_difference(B)  # Output: {1, 2, 4, 5}

Or you can use the ^ operator.

symmetric_difference_set = A ^ B  # Output: {1, 2, 4, 5}

Other Operations

Subset

Checks if a set is a subset of another set.

C = {1, 2}
print(C.issubset(A))  # Output: True

Superset

Checks if a set is a superset of another set.

print(A.issuperset(C))  # Output: True

Disjoint Sets

Checks if two sets have no elements in common.

print(A.isdisjoint(B))  # Output: False

Advanced Set Operations

Using Set Comprehensions

Similar to list comprehensions, set comprehensions are also available.

squared = {x*x for x in {1, 2, 3, 4}}  # Output: {1, 4, 9, 16}

Frozen Sets

These are immutable versions of sets that can be used as keys in dictionaries.

frozen = frozenset([1, 2, 3, 4])

Real-world Applications

  • Removing duplicates from a list.
  • Membership testing.
  • Data filtering and reduction.
  • Graph theory applications like network algorithms.

Conclusion

The set data structure in Python provides a wide array of functionalities. This enables developers to write efficient code for a multitude of applications, ranging from data science to web development. It is an indispensable tool for any Python programmer and warrants a comprehensive understanding.

Understanding the underlying set operations can go a long way in enhancing the efficiency of your Python code. Whether you are a beginner just starting out or a seasoned developer looking to optimize your code, mastering Python sets is a skill that will undoubtedly come in handy.

Leave a Reply