3D-изображение океанического течения с базовой картой - PullRequest
Новая
       35

3D-изображение океанического течения с базовой картой

1 голос
/ 30 марта 2019

Я пытаюсь построить океанское течение с помощью трехмерного объекта и базовой карты, но всякий раз, когда я пытался включить базовую карту, он отображает только данные о поверхности, а не данные о поверхности. Обычная опция манипуляции стрелками с помощью matplotlib quiver здесь не очень хорошо работает, поэтому предстоящий сюжет не хорошего качества или информативен

Я создал 3d базовую карту и считал данные из файла netcdf с выводом моей модели 

import os
import matplotlib as mpl
import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.basemap import Basemap
from mpl_toolkits.mplot3d import Axes3D
from netCDF4 import Dataset



#Reading the netcdf histoty file
in_file = Dataset('../../roms_z_his_w_t8_010.nc','r')
#temp = in_file.variables['temp'][:,:,:,:]
#salt = in_file.variables['salt'][:,:,:,:]
u = in_file.variables['u'][0,:10,:,:]
v = in_file.variables['v'][0,:10,:,:]
w = in_file.variables['w'][0,:10,:,:]
d = -(in_file.variables['depth'][:10])
lat = in_file.variables['lat'][:]
lon = in_file.variables['lon'][:]

x,y,z =  np.meshgrid(lat,d,lon)

#Create a 3d normal figure
fig = plt.figure(figsize=(16,14))
ax = fig.gca(projection='3d')

#Draw the earth map using Basemap
# Define lower left, uperright lontitude and lattitude respectively
extent = [75, 100, 5, 25]
# Create a basemap instance that draws the Earth layer
bm = Basemap(llcrnrlon=extent[0], llcrnrlat=extent[2],
             urcrnrlon=extent[1], urcrnrlat=extent[3],
             projection='cyl', resolution='l', fix_aspect=False, ax=ax)
# Add Basemap to the figure
ax.add_collection3d(bm.drawcoastlines(linewidth=0.25))
ax.add_collection3d(bm.drawcountries(linewidth=0.35))
ax.view_init(azim=300, elev=50)
ax.set_xlabel('Longitude (°E)', labelpad=20)
ax.set_ylabel('Latitude (°N)', labelpad=20)
ax.set_zlabel('Depth (m)', labelpad=20)
# Add meridian and parallel gridlines
lon_step = 5
lat_step = 5
meridians = np.arange(extent[0], extent[1] + lon_step, lon_step)
parallels = np.arange(extent[2], extent[3] + lat_step, lat_step)
ax.set_yticks(parallels)
ax.set_yticklabels(parallels)
ax.set_xticks(meridians)
ax.set_xticklabels(meridians)
#ax.set_zticks(d)
#ax.set_zticklabels(d)

skip=(slice(None,None,1),slice(None,None,1))
ax.quiver(z[skip],x[skip],y[skip],u[skip],v[skip],w[skip], length=0.1, normalize=False)
plt.savefig('3dplot.png')
in_file.close()

Мне нужен трехмерный океанский поток с функцией базовой карты и размером стрелки, меняющимся в зависимости от величины или цветовой шкалы. Прямо сейчас это выглядит так:

enter image description here

1 Ответ

1 голос
/ 31 марта 2019

Попробуйте - я добавил set_zlim (0., 150), чтобы заставить вашу ось z расширяться: 

import os
import matplotlib as mpl
import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.basemap import Basemap
from mpl_toolkits.mplot3d import Axes3D
from netCDF4 import Dataset



#Reading the netcdf history file
in_file = Dataset(r'roms.nc','r')
#temp = in_file.variables['temp'][:,:,:,:]
#salt = in_file.variables['salt'][:,:,:,:]
u = in_file.variables['u'][0,:10,:,:]
v = in_file.variables['v'][0,:10,:,:]
w = in_file.variables['w'][0,:10,:,:]
d = (in_file.variables['depth'][:10])
lat = in_file.variables['lat'][:]
lon = in_file.variables['lon'][:]

in_file.close()


x,y,z =  np.meshgrid(lat,d,lon)

#Create a 3d normal figure
fig = plt.figure(figsize=(16,14))
ax = fig.gca(projection='3d')

#Draw the earth map using Basemap
# Define lower left, uperright lontitude and lattitude respectively
extent = [75, 100, 5, 25]
# Create a basemap instance that draws the Earth layer
bm = Basemap(llcrnrlon=extent[0], llcrnrlat=extent[2],
             urcrnrlon=extent[1], urcrnrlat=extent[3],
             projection='cyl', resolution='l', fix_aspect=False, ax=ax)

# Add Basemap to the figure
ax.add_collection3d(bm.drawcoastlines(linewidth=0.25))
ax.add_collection3d(bm.drawcountries(linewidth=0.35))
ax.view_init(azim=300, elev=50)
ax.set_xlabel('Longitude (°E)', labelpad=20)
ax.set_ylabel('Latitude (°N)', labelpad=20)
ax.set_zlabel('Depth (m)', labelpad=20)

# Add meridian and parallel gridlines
lon_step = 5
lat_step = 5
meridians = np.arange(extent[0], extent[1] + lon_step, lon_step)
parallels = np.arange(extent[2], extent[3] + lat_step, lat_step)
ax.set_yticks(parallels)
ax.set_yticklabels(parallels)
ax.set_xticks(meridians)
ax.set_xticklabels(meridians)
#ax.set_zticks(d)
#ax.set_zticklabels(d)

skip=(slice(None,None,1),slice(None,None,1))
ax.quiver(z[skip],x[skip],y[skip],u[skip],v[skip],w[skip], length=0.1, normalize=False)
ax.set_zlim(0., 150)
plt.savefig('3dplot.png')

Выход: 3D Currents

...