如何从纬度/经度数据到曲面图

Question

从纬度/经度数据创建曲面图，Z 轴是某个纬度/经度位置出现的次数

我曾尝试使用 pandas 创建数据透视表，但似乎无法正常工作。它们的绘图不正确。

df2 = pd.pivot_table(df, index=['latitude', 'longitude'], 
aggfunc='count')

fig = plt.figure(figsize = (18, 18))
ax = Axes3D(fig)

x = (df2['latitude'])
y = (df2['longitude'])
z = (df2['count'])

surf = ax.plot_trisurf(x, y, z, cmap=cm.jet, linewidth=0.1)
fig.colorbar(surf, aspect = 5)
plt.show()

这是数据透视表，通常我的数据只是（纬度，经度）。在支点中，我计算了某个纬度/经度位置的命中次数。我认为问题在于枢轴的索引，您可以看到“计数”高出一行。

                count
latitude longitude  
20.1    1.1        3
20.2    1.2        3
        1.3        6
20.3    1.2        5
        1.5        5
20.4    1.1        6
        1.2        6
20.5    1.3        3
        1.4        1
        1.6        4

Answer 1

我通过使用以下代码展平表格来修复它：df2_flattened = pd.DataFrame(df2.to_records())。现在绘图成功了，因为纬度、经度和计数都填得很好。