2

我对 Python 完全陌生，我完全迷路了。我的主管帮助我生成了一个脚本来查看 3D 速度模型的一些切片：

import numpy as np
import matplotlib.pyplot as plt
import yt
from yt.units import km

#Import et reshape data
d = np.genfromtxt('velocity_model.txt', delimiter=' ')
nd=22
nx=131
vel = d[:,3].reshape(nd,nx,nx)
lat = d[:,0].reshape(nd,nx,nx)
lon = d[:,1].reshape(nd,nx,nx)
dep = d[:,2].reshape(nd,nx,nx)
# When this is read into YT, depth increases along x axis, longitude increases along y axis and latitude increases along z axis, need to swap x and z and then flip z
dep=dep.swapaxes(0,2) # swap first and third dimensions: gives lon (x), lat (y), depth (z)
vel=vel.swapaxes(0,2) # swap first and third dimensions: 
lat=lat.swapaxes(0,2) # swap first and third dimensions: 
lon=lon.swapaxes(0,2) # swap first and third dimensions: 
dep=dep[:,:,::-1] # reverse z direction
vel=vel[:,:,::-1] # swap first and 2nd dimensions: 
lat=lat[:,:,::-1] # swap first and 2nd dimensions: 
lon=lon[:,:,::-1] # swap first and 2nd dimensions: 
xmin=0
xmax=289
ymin=0
ymax=289
zmin=-100
zmax=5

#Entrer dans YT
data=dict(velocity=(vel,'km/s'),latitude=(lat,'deg'),longitude=(lon,'deg'),depth=(dep,'km'))
bbox = np.array([[xmin,xmax], [ymin,ymax], [zmin,zmax]])
ds=yt.load_uniform_grid(data,vel.shape, length_unit='km', bbox=bbox)

#Off-Axis Slice
for key in ['latitude','longitude','depth','velocity'] :
     L = [0,0,1] # cutting plane=z
     slicepos=-50
     c = [(xmax-xmin)/2, (ymax-ymin)/2, slicepos]
     cut = yt.SlicePlot(ds, L, key,origin='native',center=c) #, width=(200,90,'km'))
     cut.set_log(key, False)
     cut.annotate_text([0.5,0.9],'z={:d} km'.format(slicepos),coord_system='axis')
     cut.set_cmap(field='velocity',cmap='jet_r')
     cut.save()

使用这个脚本，我想修复颜色条，因为对于每个图像，这一个都会发生变化，而且这样解释并不容易。

我试图添加这样的限制：

h=colorbar
h.Limits = [5 9]
cut.set_cmap(field='velocity',cmap='jet_r', h)

但这不是好方法。有人有想法吗？我看到了很多东西，但没有看到 cmap。

4

2 回答 2

2

您正在寻找以下set_zlim功能：

http://yt-project.org/doc/reference/api/generated/yt.visualization.plot_window.AxisAlignedSlicePlot.set_zlim.html

该set_cmap功能只允许您选择您想要的颜色图，它不允许您设置颜色图范围。你需要使用set_zlim它。这是一个示例，使用来自http://yt-project.org/data的示例数据集之一：

import yt
ds = yt.load('IsolatedGalaxy/galaxy0030/galaxy0030')
plot = yt.SlicePlot(ds, 2, 'density')
plot.set_cmap('density', 'viridis')
plot.set_zlim('density', 1e-28, 1e-25)

这将产生以下图像：

在此处输入图像描述

于 2016-05-09T15:48:38.157 回答
1

这确实是关于yt 可视化库而不是 matplotlib 本身的问题——我已经编辑了标题和标签以反映这一点。

我以前从未遇到过，但根据官方文档yt.SlicePlot，它似乎cut要么是一个对象，要么是AxisAlignedSlicePlot一个OffAxisSlicePlot对象。这两个类都有一个.set_zlim()似乎可以做你想做的事情的方法：

AxisAlignedSlicePlot.set_zlim(*args, **kwargs)

设置颜色图的比例

参数：

  • 字段：字符串

    如果 field == 'all' 则设置颜色图比例的字段适用于所有绘图。

  • zmin : 浮动

    颜色图比例的新最小值。如果为“min”，将设置为当前视图中的最小值。

  • zmax : 浮动

    颜色图比例的新最大值。如果为'max'，将设置为当前视图中的最大值。

其他参数：

  • 动态范围：浮动（默认值：无）

    图像的动态范围。如果 zmin == None，将设置 zmin = zmax / dynamic_range 如果 zmax == None，将设置 zmax = zmin * dynamic_range。指定 dynamic_range 时，默认设置 zmin = zmax / dynamic_range。

换句话说，您可能会使用：

cut.set_zlim(field='velocity', zmin=5, zmax=9)
于 2016-05-09T15:17:23.033 回答