Create a custom ufunc for complex numbers in NumPy

Last update on April 30 2025 12:54:52 (UTC/GMT +8 hours)

18. Custom Ufunc for Complex Number Operations

ufunc with Custom Data Types:

Write a NumPy program that creates a custom ufunc to operate on complex numbers in a NumPy array.

Sample Solution:

Python Code:

import numpy as np

# Define a custom function to operate on complex numbers
def custom_complex_operation(z):
    return z.real + z.imag

# Convert the custom function to a NumPy ufunc
custom_ufunc = np.frompyfunc(custom_complex_operation, 1, 1)

# Create a NumPy array of complex numbers
complex_array = np.array([1+2j, 3+4j, 5+6j])

# Apply the custom ufunc to the complex array
result_array = custom_ufunc(complex_array)

# Print the original complex array and the result array
print("Original Complex Array:")
print(complex_array)

print("\nResult Array after applying custom ufunc:")
print(result_array)

Output:

Original Complex Array:
[1.+2.j 3.+4.j 5.+6.j]

Result Array after applying custom ufunc:
[3.0 7.0 11.0]

Explanation:

Import NumPy Library:

Import the NumPy library to handle array operations.

Define Custom Function:

Create a custom Python function custom_complex_operation to operate on complex numbers. This function returns the sum of the real and imaginary parts of a complex number.

Convert to NumPy ufunc:

Use "np.frompyfunc()" to convert the custom function to a NumPy ufunc, specifying that the function takes one input and returns one output.

Create Complex Array:

Define a NumPy array complex_array containing complex numbers.

Apply Custom ufunc:

Use the custom "ufunc" to operate on the complex array, storing the results in 'result_array'.

Finally print the original complex array and the resulting array after applying the custom "ufunc" to verify the operation.

For more Practice: Solve these Related Problems:

Write a Numpy program to define a custom ufunc that computes the magnitude of complex numbers in an array.
Write a Numpy program to create a custom ufunc that applies a complex conjugate operation and then scales the result by a given factor.
Write a Numpy program to build a custom ufunc that extracts the phase angle from each element in a complex array and then normalizes it.
Write a Numpy program to develop a custom ufunc that computes the product of two complex numbers element-wise and verify with np.multiply.

Go to:

Previous: Handling NaN values in NumPy arrays using np.nan_to_num.
Next: Combine Boolean arrays using np.logical_and in NumPy.

Python-Numpy Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.