51 lines
994 B
Python
51 lines
994 B
Python
import numpy as np
|
|
from netCDF4 import Dataset
|
|
|
|
# Load the NetCDF file
|
|
file_path = 'data/MERRA2_400.inst6_3d_ana_Np.20120101.nc4'
|
|
ncfile = Dataset(file_path, 'r')
|
|
|
|
# Check the available variables in the file
|
|
print(ncfile.variables.keys())
|
|
|
|
U = ncfile.variables['T'][:]
|
|
|
|
# Check the shape of the variable
|
|
print("Shape of U:", U.shape)
|
|
|
|
# Compute the mean of the variable across all axes (for all elements in U)
|
|
U_mean = np.mean(U)
|
|
U_sum = np.sum(U)
|
|
|
|
# Print the mean
|
|
print("Mean of U:", U_mean)
|
|
|
|
print("Sum of U:", U_sum)
|
|
|
|
sumval = 0
|
|
row = U[2,20,100]
|
|
|
|
print("Shape of row", row.shape)
|
|
print("Row data for first 20 entries", row[0:5])
|
|
|
|
row = row[0:5]
|
|
print(f"{type(row)} {row.dtype}")
|
|
|
|
n = 0
|
|
for val in row:
|
|
#if not np.ma.is_masked(val):
|
|
n+=1
|
|
sumval += np.float64(val)
|
|
Mean1 = np.mean(row)
|
|
Mean2 = sumval/n
|
|
|
|
print(type(Mean1))
|
|
print(type(Mean2))
|
|
print(Mean1)
|
|
print(Mean2)
|
|
|
|
print(f"Why does {np.mean(row):.10f} not equal {sumval/n:.10f} ?!")
|
|
|
|
|
|
# Close the NetCDF file
|
|
ncfile.close() |