import numpy as np

import os

print("=== 检查NCWF数据文件 ===")

ncwf_info_path = 'data_sampled/NCWF/ncwf_info_file.npz'

if os.path.exists(ncwf_info_path):

print(f"加载文件: {ncwf_info_path}")

ncwf_info = np.load(ncwf_info_path)

print(f"文件中的键: {list(ncwf_info.keys())}")

# 检查start_time

if 'start_time' in ncwf_info:

start_time = ncwf_info['start_time']

print(f"start_time形状: {start_time.shape}")

print(f"start_time前10个条目:")

for i in range(min(10, len(start_time))):

print(f" {i}: {start_time[i]}")

# 生成对应的文件名

print("\n对应的NCWF文件名:")

for i in range(min(5, len(start_time))):

obj = start_time[i]

fname = '%d_%s_%s_%s00Z.npz'%(obj[0], str(obj[1]).zfill(2), str(obj[2]).zfill(2), str(obj[3]).zfill(2))

print(f" {fname}")

# 检查其他键

for key in ncwf_info.keys():

if key != 'start_time':

data = ncwf_info[key]

print(f"\n{key}: 形状={data.shape}, 类型={data.dtype}")

if len(data.shape) <= 2 and data.size <= 20:

print(f" 内容: {data}")

else:

print(f"文件不存在: {ncwf_info_path}")

gridded_storm_path = 'data_sampled/NCWF/gridded_storm.npz'

if os.path.exists(gridded_storm_path):

print(f"\n\n=== 检查 {gridded_storm_path} ===")

gridded_storm = np.load(gridded_storm_path)

print(f"文件中的键: {list(gridded_storm.keys())}")

for key in gridded_storm.keys():

data = gridded_storm[key]

print(f"\n{key}: 形状={data.shape}, 类型={data.dtype}")

if key == 'start_time' and len(data) > 0:

print(f" 前5个时间: {data[:5]}")

else:

print(f"\n文件不存在: {gridded_storm_path}")