开始使用免费开始使用

从 HDF5 数据集创建 Dask 数组

您需要分析过去 40 年的欧洲降雨量。我们为您提供了以 HDF5 格式存储的欧洲网格位置的月平均降雨量数据。由于该文件较大,您决定使用 Dask 来加载并处理它。

h5py 已为您导入,dask.array 已以 da 的名称导入。

本练习是课程的一部分

Python 中的 Dask 并行编程

查看课程

练习说明

  • 使用 h5py 打开 'data/era_eu.hdf5' 文件。
  • 使用 from_array() 函数将 '/precip' 变量加载为一个 Dask 数组,并将块大小设置为(12 个月、15 个纬度、15 个经度)。
  • 使用数组切片在第一个轴上每隔 12 个取一个索引——这样可以选出所有年份的 1 月数据。
  • 沿时间轴(轴 0)对 january_rainfalls 求平均,计算整个欧洲 1 月的平均降雨量。

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Open the HDF5 dataset using h5py
hdf5_file = ____.____(____)

# Load the file into a Dask array with a reasonable chunk size
precip = da.____(____, chunks=____)

# Select only the months of January
january_rainfalls = ____[____]

# Calculate the mean rainfall in January for each location
january_mean_rainfall = ____.____(axis=____)

plt.imshow(january_mean_rainfall.compute())
plt.show()
编辑并运行代码