坐标和字段值

Question

假设我已经按如下方式加载数据集：

ds = yt.load('pltxxx')

数据集包含以下字段密度，mag_vort，示踪剂，x_velocity，y_velocity

在这种情况下，可以简单地绘制出 mag_vort，即二维域中的涡度大小，方法是：

slc = yt.SlicePlot(ds, 'z', 'mag_vort')

如果我想在 txt 文件（或 numpy 数组）中导出 x 坐标、y 坐标和 vorticity_magnitude 或通过 matplotlib 散点图绘制它

plt.scatter(x_coor, y_coor, c=mag_vort)

有没有一种简单的方法可以从数据集中提取这些信息？

Answer 1

您可以使用数据对象（在本例中我们使用 all_data 数据对象）来访问 'x'、'y' 和 [=15= 的字段值] 字段：

ad = ds.all_data()
x = ad['x']
y = ad['y']
mag_vort = ad['mag_vort']

访问数据对象返回的数组是 YTArray 个实例。 YTArray 是 numpy 的 ndarray 的子类，它附加了单位。

在将这些数组传递给 matplotlib 之前，将它们转换为您想要在绘图中使用的任何单位，然后将它们转换为 numpy 数组：

x_plot = np.array(x.to('km'))
y_plot = np.array(y.to('km'))
plt.scatter(x_plot, y_plot, c=np.array(mag_vort))

Coordinates and Field values