python - 如何将方程绘制到具有重新定义索引的数组中

Question

所以我有一个等式:x^2 + y^2

目前，我可以创建一个数组来定义方程，根据输入参数计算数组并​​打印出一个数组:

def equation(x,y):
  return x**2 + y**2

def calculate(x, y, xmin, ymin):
  out = []
  for i in range(x_min, xs):
    row = []
    for j in range(y_min, ys):
      row.append(equation(i, j))
    out.append(row)
  return out

输出数组根据索引计算值输出使得 (0,0) 位于左上角。给定一个具有长度和宽度的数组，如何计算方程以使 (0,0) 居中并遵循笛卡尔平面？

Answer 1

要将数据以 0,0,0 为中心并绘制结果，您可以执行如下操作:

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

def equation(x,y):
  return x**2 + y**2

x = [(i-50)/10 for i in range(0,100,1)]
y = x
z = [equation(i, j) for i, j in zip(x, y)]

# plot the function
fig = plt.figure()
ax = fig.gca(projection='3d')
ax.scatter(x, y, z, c='r', marker='o')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')
# rotate the plot so we can see the 3 dimensions
for angle in range(0, 360):
  ax.view_init(30, angle)
  plt.draw()
  plt.pause(.001)

结果: