How can I use parallel processing with arrays in Python?

How can I use parallel processing with arrays in Python?

You can use the multiprocessing module in Python to achieve parallel processing with arrays. Would you like a code example?

Yes, please provide a code example.

Here's a simple example using multiprocessing:

import numpy as np
from multiprocessing import Pool

# Function to square elements
def square(x):
    return x * x

# Create an array
arr = np.array([1, 2, 3, 4, 5])

# Use Pool to apply the function in parallel
with Pool() as pool:
    result = pool.map(square, arr)

print(result)  # Output: [1, 4, 9, 16, 25]

What if I want to use a different function instead of squaring?

You can simply replace the square function with any other function you want to apply to the elements of the array. Just ensure the function takes a single argument.