Plot a 3d plot in Python 来自二维矩阵中的元素

Question

I would like to plot a function f(x _1, x ₂ ) of two variables x ₁ and x ₂ in a 3D plot. The function is containted in a 2D matrix, with x ₁ forming the rows and x ₂ the columns. How do I go about plotting this?

I have tried defining my function f as

x1_axis = np.arange(0, 10, 0.1)
x2_axis = np.arange(0, 10, 0.1)

f = [fun[x1, x2] for x1 in x1_axis and x2 in x2_axis]

where 'fun' is my matrix storing the values. This raises the error 'ValueError: The truth value of an array with more than one element is ambiguous.'

Is there any other way to implement this?

Answer 1

You can create a 2d function matrix, by passing a mesh into a funcion.

import numpy as np
import matplotlib.pyplot as plt

def fun(x1, x2):
    # Define your function here
    return x1 + x2

x1_axis = np.arange(0, 10, 0.1)
x2_axis = np.arange(0, 10, 0.1)

X1, X2 = np.meshgri

d(x1_axis, x2_axis)
F = fun(X1, X2)

You can than plot that using

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_surface(X1, X2, F)

plt.show()