瞬变电磁法一维正演代码

以下是一维瞬变电磁法正演代码的示例：

import numpy as np
import matplotlib.pyplot as plt

# 定义模型参数
sigma = np.array([0.01, 0.1, 1, 10])  # 电导率
rho = 1 / sigma  # 电阻率
dz = np.array([2, 4, 6, 8])  # 层厚
z = np.concatenate(([0], np.cumsum(dz)))  # 层顶和层底深度
r = np.array([0.1, 0.2, 0.3, 0.4])  # 距离

# 定义电场计算函数
def calc_e_field(r, z, rho):
    e = np.zeros_like(r)
    for i in range(len(z) - 1):
        dz = z[i+1] - z[i]
        e += rho[i] / (2 * np.pi * r) * (np.exp(-2 * np.pi * r * z[i+1] / dz) - np.exp(-2 * np.pi * r * z[i] / dz))
    return e

# 计算电场
fig, ax = plt.subplots(figsize=(8, 6))
for i in range(len(sigma)):
    rho_i = rho[i] * np.ones_like(z)
    e_field = calc_e_field(r, z, rho_i)
    ax.plot(r, e_field, label=f'sigma={sigma[i]:.2f}')
ax.set_xlabel('Distance (m)')
ax.set_ylabel('Electric field (V/m)')
ax.set_title('1D Transient Electromagnetic Forward Modeling')
ax.legend()
plt.show()

该代码定义了模型参数，包括电导率、层厚、层顶和层底深度等，并定义了计算电场的函数`calc_e_field`。在主程序中，代码通过循环遍历不同的电导率，计算并绘制相应的电场响应。最终的输出结果是一张电场随距离变化的图像。

需要注意的是，该代码只是一个简单的示例，实际应用中需要根据具体的模型和需求进行修改和优化。