计算ERA5小时数据的风速百分位

刍狗

1. import os
   
2. import numpy as np
   
3. import xarray as xr
   
4. 
   
5. data_folder1 = 'K:\\reanalysis\ERA5\\100m_speed\\100m u\\1979-2022'
   
6. data_folder2 = 'K:\\reanalysis\ERA5\\100m_speed\\100m v\\1979-2022'
   
7. 
   
8. file_names_u = os.listdir(data_folder1)
   
9. file_names_v = os.listdir(data_folder2)
   
10. 
    
11. def read_wind(file_path_u, file_path_v, lat_value, lon_value):
    
12.     with xr.open_dataset(file_path_u) as ds:
    
13.         u_wind = ds['u100'].sel(latitude=lat_value, longitude=lon_value).values
    
14.     with xr.open_dataset(file_path_v) as ds:
    
15.         v_wind = ds['v100'].sel(latitude=lat_value, longitude=lon_value).values
    
16.     return u_wind, v_wind
    
17. 
    
18. lat_value = 0
    
19. lon_value = 90
    
20. all_u_winds = []
    
21. all_v_winds = []
    
22. 
    
23. for file_name_u, file_name_v in zip(file_names_u, file_names_v):
    
24.     file_path_u = os.path.join(data_folder1, file_name_u)
    
25.     file_path_v = os.path.join(data_folder2, file_name_v)
    
26.     u_wind, v_wind = read_wind(file_path_u, file_path_v, lat_value, lon_value)
    
27.     all_u_winds.extend(u_wind)
    
28.     all_v_winds.extend(v_wind)
    
29. 
    
30. all_u_winds = np.array(all_u_winds)
    
31. all_v_winds = np.array(all_v_winds)
    
32. 
    
33. # 计算风速
    
34. all_windspeeds = np.sqrt(np.square(all_u_winds) + np.square(all_v_winds))
    
35. 
    
36. percentile_50 = np.percentile(all_windspeeds, 50)
    
37. print(f"The 50% percentile of the wind speed values is: {percentile_50}")

复制代码

我想计算ERA5小时数据每个经纬度网格点44年的百分位，我的一个u的nc文件的大小是1.4G，总共u，v共1056个文件，所以没办法直接计算。我现在的想法是把每个经纬度网格点的的百分位算出来，循环。但是我目前写的代码计算一个经纬度网格点的百分位就非常慢，有大佬能帮我看一下问题吗