使用日期的contourf图的x轴刻度标签

时间:2017-05-20 03:16:38

标签: python matlab matplotlib plot contourf

我试图创建一个我在海上收集的速度/深度数据的时间序列等高线图。原始数据采用.mat格式,我能够将其读入一个pandas数据框,并生成一些非常接近我需要的东西,除了x轴。

此代码生成此图:

fig = plt.figure()
ax = fig.add_subplot(111)

time, depth = np.meshgrid(a2,b2)
cont = ax.contourf(time, depth, c2, 50, cmap=plt.cm.get_cmap('viridis'),vmin=-1, vmax=1)

ax.set_ylabel("Depth (m)",size=35)
ax.set_xlabel("Time (date YYY/MM/DD)", size=35)
ax.tick_params(labelsize=35)
plt.gca().invert_yaxis()
plt.title("ADCP North-South Velocity", size = 50) 

cbar = fig.colorbar(cont)
cbar.set_label(label='Velocity (m/s)', size = 35)
cbar.ax.tick_params(labelsize=35) 
mpl.rcParams['figure.figsize'] = 70,35

plt.savefig('adcpV.png')
plt.show()

第一个情节

因为它是一个matlab文件,所给出的时间值是一个参考编号(即736456(自某个参考日期以来的天数是2016年5月6日)),我可以将其转换为python datetime对象:

def matlab2datetime(matlab_datenum):
    day = dt.datetime.fromordinal(int(matlab_datenum)) - dt.timedelta(days = 366)
    return day
axisdate = []
for elem in a2:
    axisdate.append(matlab2datetime(elem))

如果我将这一行添加到第一个代码块,它会产生这个图:

ax.set_xticklabels(axisdate)

第二个情节

在第一个图中,x轴的范围大约是一年,但是当我将它们更改为转换日期时,每个刻度标记显示相同的日期。参考编号都正确地转换为正确的日期,但是同一日期有大约64,000个数据点和多个数据。因此,轴日期列表中的前几千个元素是2016-05-06,依此类推,直到2017-04-19。

第一个问题:如何获得x轴的正确日期标签?

第二个问题:如何从每个日期时间对象中删除小时,分钟和秒(00:00:00),使其仅显示当天(2016-05-06)?

以下是a2,b2和c2数据的一部分:

Float64Index([736456.208333, 736456.213542,  736456.21875, 736456.223958,
              736456.229167, 736456.234375, 736456.239583, 736456.244792,
                  736456.25, 736456.255208,
              ...
              736804.703125, 736804.708333, 736804.713541,  736804.71875,
              736804.723958, 736804.729166, 736804.734375, 736804.739583,
              736804.744791,     736804.75],
             dtype='float64', length=64142)


    Int64Index([ 4,  6,  8, 10, 12, 14, 16, 18, 20, 22, 24, 26, 28, 30,     32, 34, 36,
            38, 40, 42, 44, 46, 48, 50, 52, 54, 56, 58, 60, 62, 64, 66, 68, 70,
            72, 74, 76, 78, 80, 82, 84, 86, 88, 90, 92, 94, 96],
       dtype='int64')


[[ 0.004  0.009 -0.012 ..., -0.072 -0.09  -0.098]
 [-0.022  0.013 -0.067 ..., -0.319 -0.293 -0.253]
 [-0.053  0.005 -0.017 ..., -0.335 -0.328 -0.343]
 ..., 
 [-0.065  0.059  0.05  ..., -0.036  0.018  0.009]
 [-0.041 -0.03  -0.035 ..., -0.027 -0.07  -0.046]
 [ 0.215 -0.287  0.033 ..., -0.049  0.002 -0.01 ]]

1 个答案:

答案 0 :(得分:0)

您可以使用datetime.strftime(format)格式化日期时间,更改您的函数matlab2datetime,如下所示:

def matlab2datetime(matlab_datenum):
    day = (dt.datetime.fromordinal(int(matlab_datenum)) - dt.timedelta(days = 366)).strftime("%Y-%m-%d")
    return day
axisdate = []
for elem in a2:
    axisdate.append(matlab2datetime(elem))