使用 3d np 数组数据的 python 2d 轮廓图

Question

我这里有问题。

我的数据是一个 3d 形状的 np 数组 (256, 256, 1) = (x坐标，y坐标，压力值) 现在我想用这个 np 数组绘制一个等高线图。

但问题是我的数据的形状不适合 plt.contourf

知道如何在将数据输入到 contourf 之前对其进行预处理吗？

Answer 1

由于每个位置 [M,N] 都有一个奇异值，因此您可以简单地挤出该维度并将其用形状为 [M,N] 的矩阵表示。

data = data.squeeze(2)
plt.contourf(data)

压缩后的数组和原始数组包含完全相同的数据，只是表示方式略有不同。