Question

我有一个数据框，索引为日期时间，列为深度。我想绘制一个轮廓图，如下图所示。有什么想法我应该怎么做吗？我尝试使用plt.contour()函数，但我认为我必须首先整理数据数组。我不确定这部分。

我的数据框示例：

                     -1.62  -2.12  -2.62  -3.12  -3.62  -4.12  -4.62  -5.12

Datetime                                                                      
2019-05-24 15:45:00   4.61   5.67   4.86   3.91   3.35   3.07   3.03   2.84   
2019-05-24 15:50:00   3.76   4.82   4.13   3.32   2.84   2.40   2.18   1.89   
2019-05-24 15:55:00   3.07   3.77   3.23   2.82   2.41   2.21   1.93   1.81   
2019-05-24 16:00:00   2.50   2.95   2.63   2.29   1.97   1.73   1.57   1.48   
2019-05-24 16:05:00   2.94   3.62   3.23   2.82   2.62   2.31   2.01   1.81   
2019-05-24 16:10:00   3.07   3.77   3.23   2.82   2.51   2.31   2.10   1.89   
2019-05-24 16:15:00   2.71   3.20   2.86   2.70   2.51   2.31   2.18   1.97   
2019-05-24 16:20:00   2.50   3.07   2.86   2.82   2.73   2.50   2.37   2.22   
2019-05-24 16:25:00   2.40   3.20   3.10   2.93   2.73   2.50   2.57   2.84   
2019-05-24 16:30:00   2.21   2.95   2.86   2.70   2.73   2.72   2.91   3.49   
2019-05-24 16:35:00   2.04   2.72   2.63   2.59   2.62   2.72   3.03   3.35   
2019-05-24 16:40:00   1.73   2.31   2.33   2.39   2.62   2.95   3.57

我想要的情节示例：

对于plt.contour()中的X Y Z输入，我想找出它需要什么数据结构。它说它需要2D数组结构，但是我很困惑。如何在当前数据帧中得到它？

Answer 1

我已经解决了。请注意，X（tt2）-时间输入和Y（depth）-深度输入必须匹配Z（mat2）矩阵尺寸才能使plt.contourf正常工作。我意识到plt.contourf会生成我想要的图像，而不是仅绘制轮廓线的plt.contour。

我的代码示例： tt2 = [...] 深度= [...]

plt.title('SSC Contour Plot')
fig=plt.contourf(tt2,depth,mat2,cmap='jet',levels= 
[0,2,4,6,8,10,12,14,16,18,20,22,24,26], extend= "both")
plt.gca().invert_yaxis() #to flip the depth from shallowest to deepest
plt.gca().xaxis.set_major_formatter(mdates.DateFormatter('%d/%m/%Y %H:%M:%S'))
#plt.gca().xticklabels(tt)
cbar = plt.colorbar()
cbar.set_label("mg/l")
yy = len(colls2)
plt.ylim(yy-15,0) #assumming last 10 depth readings have NaN
plt.xlabel("Datetime")
plt.xticks(rotation=45)
plt.ylabel("Depth (m)")
plt.savefig(path+'SSC contour plot.png') #save plot 
plt.show()

Example of plot produced

以x轴为日期时间，y轴为深度的数据框绘制等高线图

1 个答案: