Python整理PEER所下载的地震波源数据——提取地震波至txt+生成地震波反应谱

导语

在进行动力时程分析前，选取合适的地震波是必须的。当前最热门的地震波下载网站当属PEER（太平洋地震工程研究中心）。从此网站下载地震波以后，往往需要将地震波数据进行二次转化，一方面得到用于导入其他软件的特种格式，另一方面生成地震波的反应谱。得到反应谱并不难，但是如何及时得到反应谱而略掉繁琐的‘点、点、点~’呢？本篇文章将解决此问题。

gitee代码托管

全部功能实现代码已托管至我的gitee账号MY，如有代码问题，可在gitee中提交，或与我联系。对于地震波数据整理所需的关键函数已全部展示于上，亦可根据提示编辑属于自己的实现程序。
创作不易，欢迎关注、点赞、转发、支持，谢谢！

一、实现目标

本篇文章的前提是处理从PEER上下载的地震波，目标是将所下载的地震波的地震波数据（存于txt中）、dt（时间步长）、ntps（时间步）、加速度反应谱四类信息进行处理。得到的结果，将地震波数据按不同时间步长整理的文件夹；含有地震波名称、dt、npts、加速度谱信息的表格。

运行效果

1 已下载的文件

为了展示，此处只下载了两条地震波，并在此目录文件内及下一级子目录文件内将地震波进行解压。可以看到，在这种状态下，当每级目录下的地震波若有100条时，或者，一条一条下载所得的地震波，处理起来的工程量不小。虽然能处理地震波的软件不少，但是地震波导入以后再依次复制出目标信息的方式非常的繁琐。

2 实现的效果

运行后，会将处理后的信息进行汇总于“处理后输出汇总”文件夹内。

“处理后输出汇总”文件夹，是根据地震波所在的目录，将地震波按目录重新创建文件夹，并将其按时间步长类型进行整理。“地震波信息汇总.xlsx”表格保存有每级目录下的地震波的名称、dt、npts、加速度谱。

二、代码过程

1 需要的库

import os
import glob
import openpyxl
from pyexcel.cookbook import merge_all_to_a_book as sss
import math

2 主要函数：共5个

函数1：得到‘.py’程序所在的文件目录及下一级子目录的路径列表a，返回列表a

def file_name(file_dir): # 将目标路径和它下一级的子目录文件夹路径全部输出至列表
    a1 = 1
    a = []
    for root, dirs, files in os.walk(file_dir):
        if a1 == 1:
            b = dirs
            a.append(file_dir)
            for i in range(0,len(b)):
                h = file_dir + '\\' + b[i]
                a.append(h)
            break
    return a

函数2：将每一路径下的‘.csv’文件转为‘.xlsx’（‘.csv’文件存有地震波名称，此步转化方便将地震波名称从‘.xlsx’中提取出来）

def csv_to_xlsx(path_1):  # path_1为列表
    h1 = file_name(path_1)
    bb1 = 1
    print('共有' + str(len(h1)) + '个含有' + '.scv' + '文件夹目录')
    for i in range(0,len(h1)):
        print('现在处理第' + str(bb1) + '个文件夹内的所有'+ ''''