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

Question

我想 plot a function f(x _1, x ₂ ) 的两个变量 x ₁和 x ₂在 3D plot 中。function 包含在二维矩阵中，x ₁构成行，x ₂构成列。 我 go 如何绘制此图？

我尝试将我的 function f 定义为

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]

其中“乐趣”是我存储值的矩阵。 这会引发错误“ValueError：具有多个元素的数组的真值不明确。”

还有其他方法可以实现吗？

Answer 1

您可以通过将网格传递给函数来创建 2d function 矩阵。

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)

你可以比 plot 使用

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

plt.show()