如何将函数矩阵f逐元素应用到数值矩阵x？附示例代码

Alright, let's tackle this problem step by step. You've got a matrix where each element is a function (f) and a numeric matrix (x), and you want to apply each function in f to the corresponding element in x—that's a common use case when working with function matrices. Here are a few solid ways to do this:

Method 1: Nested Loops (Intuitive & Easy to Debug) #

If you prefer clarity over brevity, nested loops are a great starting point. You'll iterate over each row and column index, apply the corresponding function to the matching element in x, and build your result matrix:

import numpy as np

# Define your functions
def f11(x): return 1
def f12(x): return x+1
def f21(x): return np.log(x)
def f22(x): return np.exp(x)

# Create the function matrix and numeric matrix
f = np.matrix([[f11,f12],[f21,f22]])
x = np.matrix([[10,5],[3,8]])

# Initialize an empty result matrix with the same shape as x
result = np.zeros_like(x, dtype=np.float64)

# Iterate over each element
for i in range(x.shape[0]):
    for j in range(x.shape[1]):
        # Apply the function at (i,j) to the value at x[i,j]
        result[i,j] = f[i,j](x[i,j].item())  # .item() converts numpy scalar to Python float (optional but safe)

print(result)

Running this will give you:

[[  1.          6.        ]
 [  1.09861229 2980.95798704]]

Method 2: List Comprehension (Concise & Pythonic) #

For a more compact approach, you can use nested list comprehensions to build the result in one line, then convert it back to a NumPy matrix:

import numpy as np

# Reuse the same function definitions, f, and x from above
result = np.matrix([
    [f[i,j](x[i,j].item()) for j in range(x.shape[1])]
    for i in range(x.shape[0])
])

print(result)

This will produce the exact same output as the loop method, but with less boilerplate code.

Key Notes for Complex Functions #

• This approach works for any function that accepts a single numeric input and returns a numeric output—whether it's a simple arithmetic function, a NumPy mathematical function, or a custom complex function you've written.
• Using .item() to extract the scalar value from x[i,j] is optional for most functions (since NumPy scalars behave like Python floats in most cases), but it can prevent unexpected behavior if your custom functions expect pure Python numeric types.

内容的提问来源于stack exchange，提问作者user9187374