时间戳

首页 > 解决方案 > yt ProjectionPlot 的注解选项

python - yt ProjectionPlot 的注解选项

问题描述

我是 yt 的新手，我想对我得到的 yt ProjectionPlot 输出进行一些重大调整。到目前为止，我有：

import yt
ds = yt.load("/path/to/data")
prj = yt.ProjectionPlot(ds, 'z', 'Tmean', method='integrate', weight_field='rho mean')
prj.save('Tmean_prj.png')

附加了“Tmean_prj.png”。我的第一个问题是 matplotlib 命令是否与 yt 完全兼容。然后我猜它会变得非常容易。

我想在此输出中更改以下内容：

(1) 使用派生变量进行投影，即不包含在数据集中，但可以从数据集中包含的变量推导出来的变量。

（2）由于有一个“喷气机”进入域（目前是那个紫色的东西），我想将（0,0）与紫色管道的中心和它的上边缘对齐。yt 似乎知道域尺寸和分辨率。

(3) 在大约 325 微米的（当前）x 处切割域。

(4) 将刻度由微米改为米，并以常数刻度

(5) 从绘图中分离颜色条（添加一些空白）并调整颜色和缩放比例，还将“Tmean”标签修改为垂直并位于颜色条的顶部。

在此处输入图像描述

标签： pythonmatplotlibyt-project

解决方案

针对问题 (1)，此方法有效在此处输入图像描述：

import yt
from yt.units import dimensions
from yt.units.yt_array import YTArray
import numpy as np

spNms  = ['H2', 'H', 'O2', 'OH', 'H2O', 'HO2', 'H2O2', 'CH3', 'CH4', 'CO', 'CO2', 'CH2O', 'C2H2', 'C2H4', 'C2H6', 'NH3', 'NO', 'HCN', 'N2']

HfO    = [0.0, 216269222.374, 0.0, 2292826.50902, -13423893.4949, 316904.97153, -4001375.87653, 9689965.02785, -4668373.7711, -3946437.03767, \
         -8941308.52674, -3859845.44206, 8707735.03834, 1870243.41835, -2788439.10336, -2695065.81111, 3009074.80936, 4937073.18373, 0.0]

omgdot = ['H2_reaction_rate mea', 'H_reaction_rate mean', 'O2_reaction_rate mea', 'OH_reaction_rate mea', 'H2O_reaction_rate me', 'HO2_reaction_rate me', \
          'H2O2_reaction_rate m', 'CH3_reaction_rate me', 'CH4_reaction_rate me', 'CO_reaction_rate mea', 'CO2_reaction_rate me', 'CH2O_reaction_rate m', \
          'C2H2_reaction_rate m', 'C2H4_reaction_rate m', 'C2H6_reaction_rate m', 'NH3_reaction_rate me', 'NO_reaction_rate mea', 'HCN_reaction_rate me', \
          'N2_reaction_rate mea']

#print 'HfO[16]',HfO[16]
#print 'omgdot[16]',omgdot[16]

#print 'HfO.shape',HfO.shape
#print 'omgdot.shape',omgdot.shape

def _HRR(field, data):
    #data._debug()
    Nx = data['Tmean'].shape
    HRR=YTArray(np.zeros(Nx))
    i=0
    for sp in spNms:
        HRR=HRR+(HfO[i]*data[omgdot[i]])
        i=i+1
    HRR = -HRR
    return HRR

yt.add_field(('gas','HRR'), function=_HRR)

ds = yt.load("/data1/ahoffie/iw-dm-4/HighTempLowNOx/JICF_EXP/Scaled_GT_run_12mmPre_r2/out_jicf/post/merged_stats_00027_00034")

prj = yt.ProjectionPlot(ds, 'z', 'HRR', method='integrate', weight_field='rho mean')

prj.save('HRR_prj.png')

推荐阅读