定义一个函数select_data()

  1. 定义变量dbz_dir_path,赋值为字符串'data/NJU_CPOL_update2308/dBZ/3.0km/'。
  2. 定义变量kdp_dir_path,赋值为字符串'data/NJU_CPOL_update2308/KDP/3.0km/'。
  3. 定义变量zdr_dir_path,赋值为字符串'data/NJU_CPOL_update2308/ZDR/3.0km/'。
  4. 定义变量data_select,赋值为字符串'data/select.txt'。
  5. 打开文件data_select_file,以追加模式写入数据。
  6. 获取目录列表dir_list,使用os.listdir(dbz_dir_path)
  7. 遍历dir_list中的每个目录,使用变量名dir
  8. 构建dbz_data_path,为dbz_dir_path + dir
  9. 获取dbz_data_path目录下的文件列表dbz_list,使用os.listdir(dbz_data_path)
  10. 初始化变量i为0。
  11. 初始化空列表data_list
  12. 遍历dbz_list的元素,使用range(len(dbz_list))循环。
  13. dbz_data_path + '/' + dbz_list[i]加载为Numpy数组,并添加到data_list中。
  14. 如果(len(dbz_list) - i) < (20 - len(data_list))为真,则跳出循环。
  15. 如果len(data_list)等于20,则继续执行以下步骤:
  16. data_list转换为Numpy数组。
  17. 如果(np.sum(data_list == 0)) < 0.8 * 20 * 256 * 256为真,则执行以下步骤:
  18. data_select_file中写入字符串' 3.0km,dir,dbz_list[i + 1 - 20]',其中dirdbz_list[i + 1 - 20]为对应的变量值。
  19. 打印字符串'dir: dir, frame: dbz_list[i + 1 - 20]'。
  20. data_list重置为空列表。
  21. 继续下一次循环。
  22. 否则,将i减去19。
  23. data_list重置为空列表。
  24. 继续下一次循环。

原文地址: http://www.cveoy.top/t/topic/bjWV 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录