如何转换为数组？Matplotlib等高线图为何仅显示一半？

以下是Python中几种常见场景的转换方法：

• 列表/元组转数组：直接用numpy.array()转换

  import numpy as np
  data_list = [1,2,3,4]
  arr = np.array(data_list)
  

• Pandas数据转数组：DataFrame或Series可通过.to_numpy()转换

  import pandas as pd
  df = pd.DataFrame({'a':[1,2], 'b':[3,4]})
  arr = df.to_numpy()
  

• 文本/CSV数据转数组：用numpy.loadtxt()读取后直接得到数组

  arr = np.loadtxt('data.csv', delimiter=',')
  

• 自定义结构数据转数组：先提取目标字段为列表，再转数组

  data_dict = {'x':[1,2], 'y':[3,4]}
  arr = np.array(list(data_dict.values())).T
  

你的代码存在两个核心问题：

1. 网格范围未覆盖函数关键区域
   你定义的函数F = (x1 - 2) ** 2 + 2 * (x2 - 3) ** 2最小值点在(2, 3)，但设置的x = np.linspace(-1,1,100)仅覆盖x ∈ [-1,1]，完全在最小值点左侧，导致所有计算出的F值只对应椭圆等高线的左半部分，无法显示完整形态。
2. y轴数据依赖x轴，未生成独立网格
   通过y = 2 * x + 3 + np.random.rand(len(x))生成的y是x的函数，并非独立的轴范围，这种方式下的网格无法覆盖连续的y轴区域，进一步限制了图形的显示范围。

修正后的代码 #

import numpy as np
from matplotlib import pyplot as plt

# 扩大x范围至包含最小值点x=2，生成独立的y轴范围
x = np.linspace(-1, 3, 100)
y = np.linspace(1, 5, 100)
x1, x2 = np.meshgrid(x, y)

F = (x1 - 2) ** 2 + 2 * (x2 - 3) ** 2

# 3D曲面图：将等高线投影至曲面底部，避免重叠
fig = plt.figure()
surf = fig.add_subplot(1,1,1, projection='3d')
surf.plot_surface(x1, x2, F, cmap='viridis')
surf.contour(x1, x2, F, 20, offset=np.min(F), cmap='coolwarm')
plt.show()

# 平面等高线图：添加标签增强可读性
fig, ax = plt.subplots()
contour = ax.contour(x1, x2, F, 20, cmap='coolwarm')
ax.clabel(contour, inline=True, fontsize=8)
plt.show()

修正说明 #

• 调整x、y范围至包含函数最小值点，确保网格覆盖完整的等高线区域；
• 生成独立的y轴范围，让网格覆盖矩形区域，保证曲面和等高线完整显示；
• 给3D图的等高线设置offset参数，避免与曲面对象重叠导致显示异常。

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