拿地震数据做网络训练时,需要生成相关的label.本文将excel表格数据转化为sac格式label。
import os
import pandas as pd
import numpy as np
from obspy import Trace
from tqdm import tqdm
import matplotlib.pyplot as plt
import obspy
if __name__ == '__main__':
for i in tqdm(range(10000),desc = 'processing'):
file = pd.read_excel('/Users/xichengpeng/work/data/xlsx/Z1.xlsx')
win = np.zeros([len(file),2])
win[:,0] = np.array(file[0])
win[:,1] = np.array(file[1])
pre = np.zeros([4992,520])
for j in range(0,8):
# i = 519
pre[int(win[i*8+j,0]):int(win[i*8+j,1]),1] = 1
time = np.arange(4992) * 0.0006
sacfile = Trace()
sacfile.data = pre[:,1]
sacfile.write('/Users/xichengpeng/work/data/label/Z1_label/' + str(i) + '.sac')
# plt.plot(time,pre,'r')