使用python读写segy文件

SEGY是地震勘探中常用的数据格式，Python中可以使用ObsPy库来读写SEGY文件。

首先需要安装ObsPy库，可以使用pip命令进行安装：

pip install obspy

然后就可以使用以下代码来读取SEGY文件：

from obspy.io.segy.core import _read_segy

filename = "example.sgy"
stream = _read_segy(filename, headonly=True)
print(stream)

其中，`filename`为SEGY文件名，`_read_segy`函数用于读取SEGY文件，`headonly=True`表示只读取文件头信息，如果需要读取文件中的数据，则将`headonly`参数设置为`False`。

如果要写入SEGY文件，可以使用以下代码：

from obspy.core import Trace, Stream
from obspy.io.segy.segy import SEGYTraceHeader, SEGYBinaryFileHeader

# 创建SEGY头信息
binary_file_header = SEGYBinaryFileHeader()
trace_header = SEGYTraceHeader()

# 创建Trace对象，包含数据和头信息
tr = Trace(data=[1, 2, 3, 4], header=trace_header)

# 创建Stream对象，包含多个Trace
st = Stream(traces=[tr])

# 写入SEGY文件
st.write("example.sgy", format="SEGY", data_encoding=1, byteorder=">")

其中，`Trace`对象包含数据和头信息，可以使用`SEGYTraceHeader`类创建头信息，`Stream`对象包含多个`Trace`，可以使用`SEGYBinaryFileHeader`类创建文件头信息。`write`函数用于将`Stream`对象写入SEGY文件，`format`参数指定文件格式，`data_encoding`参数指定数据编码方式，`byteorder`参数指定字节顺序。