Python中的海洋绘图

时间:2016-09-05 14:34:41

标签: python python-2.7 python-3.x numpy matplotlib

我正在绘制图表。我想在图形的颜色条上有一系列的值" U_velocity"和" U_shear_velocity"从-0.3到0.3。此外,我试图使U轴和V轴剪切速度图的x轴范围从0到12.5,但没有任何效果,而不是我有速度的意义。我怎么能这样做,请帮助我。

from netCDF4 import *
import matplotlib as mp
import numpy as np

#import matplotlib.pyplot as plt
import pylab as plt


#%%

file = "/home/vlad/Desktop/task/Untitled Folder/result.nc" 


ncdata = Dataset(file, 'r')

u = np.squeeze(ncdata.variables['u'][:])
v = np.squeeze(ncdata.variables['v'][:])
z = np.squeeze(ncdata.variables['z'][:])

time = ncdata.variables['time'][:]/3600

ncdata.close()

u_mean = np.mean(u[0:100,:],0)
z_mean = np.mean(z[0:100,:],0)
v_mean = np.mean(v[0:100,:],0)

u_mean_10 = u[900:1000,:]
v_mean_10 = v[900:1000,:]
z_10 = np.mean(z[900:1000,:],0)

time_10 = time[900:1000] - time[900]


T = len(time_10)
L = len(z_10)

fig = plt.figure(6)
plt.pcolormesh(time_10,z_10,u_mean_10.T)
plt.xlim([0, time_10[-1]])
fig.suptitle('U_velocity', fontsize=25)
plt.xlabel('time', fontsize=20)
plt.ylabel('depth(m)', fontsize=20)
plt.colorbar()
plt.show()

shear_u_mean_10 = np.zeros([T,L])

for t in range(T):
    for i in range(L-1):
        tmp=(u_mean_10[t, i+1]-u_mean_10[t, i])/(z_10[i+1]-z_10[i])
        tmp_depth = 0.5 * (z_10[i+1]+z_10[i])
        shear_u_mean_10[t,i] = tmp

fig = plt.figure(10)
plt.pcolormesh(time_10/3600,z_10, shear_u_mean_10.T)
plt.xlim([0, time_10[-1]/3600])
plt.colorbar()
#plt.ylim([-30, -25])
fig.suptitle('U_shear velocity', fontsize=25)
plt.xlabel('time', fontsize=20)
plt.ylabel('depth(m)', fontsize=20)
plt.show()


shear_v_mean_10 = np.zeros([T,L])

for t in range(T):
    for i in range(L-1):
        tmp=(v_mean_10[t, i+1]-v_mean_10[t, i])/(z_10[i+1]-z_10[i])
        tmp_depth = 0.5 * (z_10[i+1]+z_10[i])
        shear_v_mean_10[t,i] = tmp

fig = plt.figure(11)
plt.pcolormesh(time_10/3600,z_10, shear_v_mean_10.T)
plt.xlim([0, time_10[-1]/3600])
plt.colorbar()
#plt.ylim([-30, -25])
fig.suptitle('V_shear velocity', fontsize=25)
plt.xlabel('time', fontsize=20)
plt.ylabel('depth(m)', fontsize=20)
plt.show()



fig = plt.figure(7)
plt.pcolormesh(time_10,z_10,v_mean_10.T)
plt.xlim([0, time_10[-1]])
fig.suptitle('V_velocity', fontsize=25)
plt.xlabel('time', fontsize=20)
plt.ylabel('depth(m)', fontsize=20)
plt.colorbar()
plt.show()

1 个答案:

答案 0 :(得分:0)

使用代码墙,未知文件result.nc的引用以及一些不相关且相当具体的问题,这不是一个简单的问题。以下内容可能有所帮助:

可以通过将vmin=-0.3vmax=0.3传递给pcolormesh来设置颜色栏范围。

要限制时间范围,可以使用数组切片(例如time[time<12.5], u[time<12.5])。

对于您的数据,速度肯定只是speed = np.sqrt(np.power(u,2) + np.power(v,2)

如果您需要进一步的帮助,请提供最小,完整且可验证的示例。