np.frompyfunc() for ufunc creation in NumPy - Mini Project: Build & Apply

Create a Custom Universal Function with np.frompyfunc()

📖 Scenario: Imagine you work in a small bakery that tracks daily sales of different types of bread. You want to quickly calculate the total revenue from each type by multiplying the number of breads sold by their price. You want to create a reusable function that works on arrays of sales and prices easily.

🎯 Goal: You will create a custom universal function (ufunc) using np.frompyfunc() that multiplies two numbers. Then, you will apply this ufunc to arrays of sales and prices to get total revenue per bread type.

📋 What You'll Learn

Create a Python function that multiplies two numbers

Use np.frompyfunc() to convert this function into a ufunc

Apply the ufunc to two numpy arrays representing sales and prices

Print the resulting array of total revenues

💡 Why This Matters

🌍 Real World

Custom ufuncs let you apply your own Python logic efficiently on numpy arrays, useful in data analysis and scientific computing.

💼 Career

Understanding ufunc creation helps in optimizing data processing tasks and extending numpy's capabilities in data science roles.

1

Create a Python function to multiply two numbers

Write a Python function called multiply that takes two arguments a and b and returns their product.

NumPy

# Define the multiply function
# Your code here

Need a hint?

Use def multiply(a, b): and return a * b.

2

Create a ufunc from the multiply function using np.frompyfunc()

Import numpy as np. Then create a ufunc called multiply_ufunc by using np.frompyfunc() on the multiply function with 2 input arguments and 1 output.

NumPy

def multiply(a, b):
    return a * b

# Import numpy and create multiply_ufunc using np.frompyfunc
# Your code here

Need a hint?

Use np.frompyfunc(multiply, 2, 1) to create the ufunc.

3

Create numpy arrays for sales and prices and apply the ufunc

Create two numpy arrays: sales = np.array([10, 15, 7]) and prices = np.array([2.5, 3.0, 4.0]). Then use multiply_ufunc on sales and prices to get total_revenue.

NumPy

import numpy as np

def multiply(a, b):
    return a * b

multiply_ufunc = np.frompyfunc(multiply, 2, 1)

# Create sales and prices arrays
# Use multiply_ufunc on sales and prices to get total_revenue
# Your code here

Need a hint?

Use np.array() to create arrays and call multiply_ufunc(sales, prices).

4

Print the total revenue array

Print the total_revenue array to see the result of multiplying sales by prices.

NumPy

import numpy as np

def multiply(a, b):
    return a * b

multiply_ufunc = np.frompyfunc(multiply, 2, 1)

sales = np.array([10, 15, 7])
prices = np.array([2.5, 3.0, 4.0])
total_revenue = multiply_ufunc(sales, prices)

# Print total_revenue
# Your code here

Need a hint?

Use print(total_revenue) to display the array.