python处理地震波文件

zb580504

自己专业方向是抗震，昨天一个同学问我要一条地震波，无奈自己以前用的地震波都是师兄给的，已经处理好了。
大家都知道一般可以在PEER上下载地震波文件。但原始文件是多行多列的，我们一般习惯用那种单列的数据。
当然可以将多行多列数据拷到Excel中，用OFFSET函数将其变成单列数据。但这样感觉还是稍微麻烦了一点。
正好最近在学python，看了很多别人编的代码，很少自己动手实践。前段时间写了个py文件读取abaqus的odb文件
的时程曲线，无奈数据点太多，可能方法不当，代码效率不是很高。
一直就比较羡慕别人用编程语言操作文本，今天算是自己头一次用python操作文本文件。对于高手来说，我写出的代码
必然会有很多的不足之处，欢迎提出问题。
PEER上下载的地震波文件一般是这样的


前四行数据是对该地震波的一些相应情况的说明，从第五行开始是地震波记录，每行5个数据，文件扩展名是AT2。

编写的代码如下：

1. #!/user/bin/python
   
2. # -*-coding:utf-8 -*-
   
3. def dizhenbochuli (filenamein,filenameout,dt):
   
4.   f=open(filenamein,'r')
   
5.   f1=open(filenameout,'w')
   
6.   lines=f.readlines()
   
7.   lines=[line.split() for line in lines]
   
8.   liness=lines[4:]
   
9.   a=[]
   
10.   j=0
    
11.   delta=float(dt)
    
12.   for line in liness:
    
13.     for i in range(len(line)):
    
14.      j=j+1
    
15.      a.append(str((j-1)*delta)+','+str(line[i])+'\n')
    
16.   f1.writelines(a)
    
17.   f1.close()
    
18.   f.close()
    
19. if __name__=='__main__':
    
20.   filenamein=raw_input('input filename:')
    
21.   filenameout=raw_input('output filename:')
    
22.   dt=raw_input('地震波时间间隔dt为:')
    
23.   dizhenbochuli(filenamein,filenameout,dt)
    

复制代码

代码运行如下图所示：



这样就生成了需要的地震波文件，我将第一列设成了时间，第二列地震波记录，中间用逗号隔开：
如下所示：


附上PY文件！

zb580504

近日闲来无聊，正好最近正在学OpenSees ，发现OpenSees中的地震波文件是那种一行五个数据点的，于是把前面的代码稍微改了下：

1. 
   
2. #!/user/bin/python
   
3. # -*-coding:utf-8 -*-
   
4. #该函数将一般一行5个数据点的地震波文件转化成第一列时间，第二列数据点，中间逗号隔开的文件！
   
5. def dizhenbochuli (filenamein,filenameout,dt):
   
6. f=open(filenamein,'r')
   
7. f1=open(filenameout,'w')
   
8. lines=f.readlines()
   
9. lines=[line.split() for line in lines]
   
10. #liness=lines[4:]
    
11. a=[]
    
12. j=0
    
13. delta=float(dt)
    
14. for line in lines:
    
15.   for i in range(len(line)):
    
16.    j=j+1
    
17.    a.append(str((j-1)*delta)+','+str(line[i])+'\n')
    
18. f1.writelines(a)
    
19. f1.close()
    
20. f.close()
    
21. #该函数将上面得到的地震波文件反过来转化成一行5个数据点的地震波文件！
    
22. def OpenSeesdizhenbo (filenamein,filenameout):
    
23. f=open(filenamein,'r')
    
24. f1=open(filenameout,'w')
    
25. lines=f.readlines()
    
26. lines=[line.split(',') for line in lines]
    
27. a=[]
    
28. m=len(lines)
    
29. j=0
    
30. for i in range(m):
    
31.   for line in lines[5*i:5*i+5]:
    
32.    a.append(str(line[1]))
    
33. for line in lines[5*i+5:]:
    
34.   a.append(str(line[1]))
    
35. b=[]
    
36. for n in range(len(a)):
    
37.   if n%5!=4:
    
38.    b.append(a[n].replace('\n','')+' ')
    
39.   else:
    
40.    b.append(a[n])
    
41. f1.writelines(b)
    
42. f1.close()
    
43. f.close()
    
44. if __name__=='__main__':
    
45. filenamein=raw_input('input filename:')
    
46. filenameout=raw_input('output filename:')
    
47. dt=raw_input('地震波时间间隔dt为:')
    
48. dizhenbochuli(filenamein,filenameout,dt)

复制代码

1. #通过以下调用便可：（该文件存成dizhenbochuli.py）
   
2. from dizhenbochuli import *
   
3. dizhenbochuli(*,*,*)#转成时间间隔-数据点形式，中间逗号隔开.
   
4. OpenSeesdizhenbo(*,*)#转成一行5个数据点形式
   

复制代码

pbscut

不错，但是对于编程不熟练的童鞋，其实还有一种更简便的方法。直接在Excel里面导入Txt文件，稍加修改即可变成只有两列的数据。

zb580504

pbscut 发表于 2011-10-20 23:29
不错，但是对于编程不熟练的童鞋，其实还有一种更简便的方法。直接在Excel里面导入Txt文件，稍加修改即可变 ...

呃，这个我知道，而且我这里也有一个excel做的，但考虑到那样还得倒来倒去比较麻烦，就编了个这个来简化一下，而且最主要是联系下python，最近正在学这个。

nashone

学习一下，非常有用

zuihao

学习一下，非常有用

qi9876543

谢谢楼主无私提供程序~在此表示感谢~

aqustut

很有用的分享