如何在python中创建netCDF文件时在循环中写入变量

import netCDF4 as nc

# 创建netCDF文件
dataset = nc.Dataset('example.nc', 'w', format='NETCDF4')

# 创建维度
time_dim = dataset.createDimension('time', None)
lat_dim = dataset.createDimension('lat', 10)
lon_dim = dataset.createDimension('lon', 10)

# 创建变量
time_var = dataset.createVariable('time', 'f8', ('time',))
lat_var = dataset.createVariable('lat', 'f4', ('lat',))
lon_var = dataset.createVariable('lon', 'f4', ('lon',))
data_var = dataset.createVariable('data', 'f4', ('time', 'lat', 'lon',))

# 写入变量值
for i in range(10):
    time_var[i] = i
    lat_var[:] = range(10)
    lon_var[:] = range(10)
    data_var[i, :, :] = i

# 设置变量属性
time_var.units = 'hours since 2022-01-01 00:00:00'
lat_var.units = 'degrees_north'
lon_var.units = 'degrees_east'
data_var.units = 'm/s'
data_var.description = 'Some data'

# 关闭netCDF文件
dataset.close()