DEV Community

Abdulla Ansari
Abdulla Ansari

Posted on

20 Powerful Techniques for Writing Efficient and Readable Python Code

Python is known for its simplicity and versatility, but even seasoned developers benefit from adopting best practices that maximize performance and readability. With the rise of data science, machine learning, and web development in Python, mastering efficient code techniques has become a must to stay competitive in today’s fast-moving tech landscape. Here, we’ll dive into 20 effective techniques to improve your Python code’s performance and readability, whether you're working on a complex project or a quick automation script.

1. Use Generators to Conserve Memory

Generators are ideal for processing large data sets without using excessive memory. They yield data one piece at a time, instead of holding everything in memory. For instance, you can read a large log file line by line with a generator.

def read_large_file(file_path):
    with open(file_path, 'r') as file:
        for line in file:
            yield line.strip()
Enter fullscreen mode Exit fullscreen mode

This approach is especially useful for tasks like data processing or batch training, where working with limited memory is essential.

2. Set Default Values with .setdefault()

In cases where you need to initialize keys in a dictionary with default values, .setdefault() saves you from manual checks.

inventory = {"jeans": 500, "top": 600}
inventory.setdefault("shoes", 0)
print(inventory)
Enter fullscreen mode Exit fullscreen mode

This makes managing default values much more concise and removes the need for extra if-statements.

3. Replace if-elif Chains with a Dictionary

Using a dictionary to map functions instead of long if-elif chains makes code cleaner and more maintainable.

def start(): print("Start")
def stop(): print("Stop")
actions = {"start": start, "stop": stop}
action = "start"
actions.get(action, lambda: print("Invalid"))()
Enter fullscreen mode Exit fullscreen mode

This structure improves readability and performance, particularly in large decision trees.

4. Simplify Counting with Counter

The Counter class from the collections module is a great way to simplify counting tasks in Python, such as frequency analysis.

from collections import Counter
words = ["apple", "banana", "apple", "orange", "banana"]
counts = Counter(words)
print(counts)
Enter fullscreen mode Exit fullscreen mode

It eliminates the need for creating custom counting functions and is both efficient and easy to use.

5. Optimize Recursion with Memoization

Memoization stores results of expensive function calls, which is particularly useful in recursive algorithms like Fibonacci calculations.

from functools import lru_cache

@lru_cache(maxsize=1000)
def fibonacci(n):
    if n < 2:
        return n
    return fibonacci(n - 1) + fibonacci(n - 2)
Enter fullscreen mode Exit fullscreen mode

This approach reduces time complexity at the cost of minimal additional memory.

6. Add Flexibility with Decorators

Python decorators are useful for applying reusable functionality to multiple functions, like logging or timing without modifying core logic.

import time

def timer(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        print(f"{func.__name__} took {time.time() - start_time:.6f} seconds")
        return result
    return wrapper

@timer
def slow_function():
    time.sleep(1)

slow_function()
Enter fullscreen mode Exit fullscreen mode

7. Make Data Models Clear with dataclass

Python’s dataclass makes defining simple data models easier and more readable by automatically generating init, repr, and comparison methods.

from dataclasses import dataclass

@dataclass
class Employee:
    name: str
    id: int
    salary: float

e = Employee("Alice", 1, 50000)
print(e)
Enter fullscreen mode Exit fullscreen mode

This helps reduce boilerplate code and keeps your data structures clean and maintainable.

8. Structure Conditions with match

With Python 3.10, structural pattern matching allows you to match complex data structures without verbose if-else statements.

def describe_point(point):
    match point:
        case (0, 0):
            return "Origin"
        case (0, y):
            return f"On Y-axis at {y}"
        case (x, 0):
            return f"On X-axis at {x}"
        case (x, y):
            return f"Point at ({x}, {y})"
Enter fullscreen mode Exit fullscreen mode

9. Replace Chained and with all()

To validate multiple conditions at once, use all() to keep code concise and readable.

fields = ["name", "email", "age"]
data = {"name": "Alice", "email": "alice@example.com", "age": 25}
if all(field in data for field in fields):
    print("All fields are present")
Enter fullscreen mode Exit fullscreen mode

10. Utilize List Comprehensions

List comprehensions make loops concise and expressive, especially for simple transformations.

squares = [x ** 2 for x in range(10)]
Enter fullscreen mode Exit fullscreen mode

They are more efficient and easier to read than traditional loops.

11. Understand and Use Generator Expressions

For cases where you don’t need a list, use generator expressions for better memory efficiency.

sum_of_squares = sum(x ** 2 for x in range(1000))
Enter fullscreen mode Exit fullscreen mode

Generator expressions reduce memory usage by producing values on demand.

12. Try zip() for Parallel Iteration

The zip() function makes it easy to iterate over multiple lists in parallel.

names = ["Alice", "Bob"]
ages = [25, 30]
for name, age in zip(names, ages):
    print(f"{name} is {age} years old")
Enter fullscreen mode Exit fullscreen mode

13. Handle Files Safely with with Statements

The with statement ensures files are properly closed after their suite finishes, making it ideal for file handling.

with open("data.txt") as file:
    data = file.read()
Enter fullscreen mode Exit fullscreen mode

This simplifies resource management and minimizes potential for errors.

14. Add Safety with Type Hints

Type hints make your code more readable and help IDEs detect potential errors before runtime.

def add(a: int, b: int) -> int:
    return a + b
Enter fullscreen mode Exit fullscreen mode

Type hints improve maintainability, especially in large codebases.

15. Simplify with any() for or Conditions

To check if any condition in a list is true, any() is more concise than chained or conditions.

permissions = ["read", "write"]
if any(permission == "admin" for permission in permissions):
    print("Admin access granted")
Enter fullscreen mode Exit fullscreen mode

16. Leverage try-except-else-finally

This structure allows for cleaner error handling, with else and finally adding flexibility to manage different scenarios.

try:
    result = risky_function()
except ValueError:
    print("Invalid input")
else:
    print("Success:", result)
finally:
    print("Operation complete")
Enter fullscreen mode Exit fullscreen mode

17. Organize Data with Named Tuples

Named tuples add structure to tuples, making them more readable and self-documenting.

from collections import namedtuple

Employee = namedtuple("Employee", "name id salary")
e = Employee(name="Alice", id=1, salary=50000)
Enter fullscreen mode Exit fullscreen mode

18. Improve str Concatenation with f-Strings

f-Strings are faster and more readable than traditional concatenation methods, especially with complex expressions.

name = "Alice"
print(f"Hello, {name}!")
Enter fullscreen mode Exit fullscreen mode

19. Use itertools for Efficient Iterations

The itertools module offers efficient looping options, like generating permutations, combinations, or repeating elements.

from itertools import permutations

items = [1, 2, 3]
for perm in permutations(items):
    print(perm)
Enter fullscreen mode Exit fullscreen mode

20. Keep Code Clean with Context Managers

Custom context managers help manage resources or cleanup tasks, improving readability and safety.

from contextlib import contextmanager

@contextmanager
def open_file(name):
    file = open(name, "w")
    yield file
    file.close()
Enter fullscreen mode Exit fullscreen mode

By integrating these techniques, you can write Python code that is not only more efficient but also more readable and maintainable. Experiment with these tips, and gradually incorporate them into your everyday coding practices.

Top comments (0)