从usgs上批量下载径流数据(自用)

最新推荐文章于 2023-11-23 08:57:34 发布

usernameisnotem

最新推荐文章于 2023-11-23 08:57:34 发布

阅读量339

点赞数 1

文章标签： python 开发语言

本文链接：https://blog.csdn.net/usernameisnotem/article/details/131998276

版权

该Python代码实现了一个功能，从指定的URL模板中生成链接，用于下载1到18号文件夹中的USGS流速数据txt文件。它首先获取文件夹中的文件名，移除.txt后缀，然后构造URL请求流数据，并将内容保存为新的txt文件。

摘要由CSDN通过智能技术生成

import requests
import os

def download_txt_file(url, filename):
    response = requests.get(url, allow_redirects=True)
    with open('./data/' + filename, 'wb') as f:
        f.write(response.content)

def get_file_names_in_folder(folder_path):
    file_names = []
    for file_name in os.listdir(folder_path):
        if os.path.isfile(os.path.join(folder_path, file_name)):
                file_name = file_name.replace('.txt', '')
                file_names.append(file_name)
    return file_names

if __name__ == '__main__':
    for i in range(1, 19):
        folder_path = '/Volumes/TOSHIBAEXT/files/usgs_streamflow/' + str(i)
        file_names = get_file_names_in_folder(folder_path)
        #print(file_names)
        for file_name in file_names:
            url = 'https://waterdata.usgs.gov/nwis/dv?cb_00060=on&format=rdb&site_no={}&legacy=&referred_module=sw&period=&begin_date=2023-05-01&end_date=2023-07-29'.format(file_name)
            filename = '{}.txt'.format(file_name)
            download_txt_file(url, filename)