GemPy中3D点不显示问题的解决方案

# 创建一个新的conda环境（推荐）
conda create -n gempy_env python=3.10
conda activate gempy_env

# 在新环境中安装GemPy
pip install gempy

# 或者，如果使用pipenv
pipenv --python 3.10
pipenv install gempy

import gempy as gp
import gempy_viewer as gpv # GemPy 2.3.0及以后版本推荐使用gempy_viewer进行可视化

# 1. 创建地质模型
# 'your_model_name' 应替换为您的模型名称
geo_model = gp.create_model('your_model_name')

# 2. 初始化数据
# 'your_data' 应替换为您的地质数据，这通常是一个包含点、表面等信息的字典或DataFrame
# 例如：gp.init_data(geo_model, extent=[0,100,0,100,0,100], resolution=[50,50,50], path_to_data='path/to/your/data.csv')
# 确保您的数据中包含用于3D点绘制的观测点信息
# 此处为了示例，我们假设'your_data'已正确加载或定义
gp.init_data(geo_model,
             extent=[0, 100, 0, 100, 0, 100], # 示例范围
             resolution=[50, 50, 50],       # 示例分辨率
             )

# 3. 映射地质层序到表面
# 这一步至关重要，它告诉GemPy如何将您的数据点（例如，层位点、断层点）解释为地质表面
# 'Fault_Series' 和 'Strat_Series' 是示例的地质层序名称
# 'your_fault' 和 'your_strat' 应替换为实际的断层和地层名称（或列表）
# 例如：
# geo_model.add_surface_points(X=[...], Y=[...], Z=[...], series='Strat_Series', surface='surface1')
# geo_model.add_surface_points(X=[...], Y=[...], Z=[...], series='Fault_Series', surface='fault1')
# 然后进行映射
gp.map_stack_to_surfaces(geo_model,
                        {"Fault_Series": 'your_fault_surface_name', # 示例：'fault1'
                         "Strat_Series": ['your_strat_surface_name_1', 'your_strat_surface_name_2']}, # 示例：['layer1', 'layer2']
                         )

# 注意：在调用gp.map_stack_to_surfaces之前，您需要通过geo_model.add_surface_points()
# 或 geo_model.add_surface_orientations()等方法向模型添加实际的地质观测点数据。
# 否则，即使映射了，也没有点可以显示。

# 4. 绘制3D模型
# 确保在运行此行之前，模型中已经有数据并且已通过map_stack_to_surfaces进行了映射
gpv.plot_3d(geo_model)

import gempy as gp
import gempy_viewer as gpv
import numpy as np
import pandas as pd

# --- 1. 环境配置（请确保您的Python环境是3.10） ---

# --- 2. 准备示例数据（实际应用中，您将从文件加载数据） ---
# 假设我们有一些地层点和断层点
# 地层点数据
strat_points_df = pd.DataFrame({
    'X': [10, 20, 30, 40, 50, 60, 70, 80],
    'Y': [10, 20, 30, 40, 50, 60, 70, 80],
    'Z': [90, 80, 70, 60, 50, 40, 30, 20],
    'surface': ['Layer1', 'Layer1', 'Layer2', 'Layer2', 'Layer3', 'Layer3', 'Layer4', 'Layer4']
})

# 断层点数据
fault_points_df = pd.DataFrame({
    'X': [25, 25, 25, 75, 75, 75],
    'Y': [10, 50, 90, 10, 50, 90],
    'Z': [90, 50, 10, 90, 50, 10],
    'surface': ['Fault1', 'Fault1', 'Fault1', 'Fault2', 'Fault2', 'Fault2']
})

# --- 3. GemPy模型初始化与数据加载 ---

# 创建模型
geo_model = gp.create_model('MyGeologicalModel')

# 初始化数据：设置模型范围和分辨率
gp.init_data(geo_model,
             extent=[0, 100, 0, 100, 0, 100], # X, Y, Z 范围
             resolution=[50, 50, 50],        # 模型网格分辨率
             )

# 添加地层系列
geo_model.add_series(name='Stratigraphy_Series', order_series=gp.Series.ResolutionOrder.ASCENDING)
geo_model.add_surfaces(geo_model.series['Stratigraphy_Series'], ['Layer1', 'Layer2', 'Layer3', 'Layer4'])

# 添加断层系列
geo_model.add_series(name='Fault_Series', order_series=gp.Series.ResolutionOrder.ASCENDING,
                     structural_frame=True,
                     nelements=2) # 假设有两个断层
geo_model.add_surfaces(geo_model.series['Fault_Series'], ['Fault1', 'Fault2'])


# 将观测点数据添加到模型中
# 添加地层点
for index, row in strat_points_df.iterrows():
    geo_model.add_surface_points(X=row['X'], Y=row['Y'], Z=row['Z'],
                                 series='Stratigraphy_Series', surface=row['surface'])

# 添加断层点
for index, row in fault_points_df.iterrows():
    geo_model.add_surface_points(X=row['X'], Y=row['Y'], Z=row['Z'],
                                 series='Fault_Series', surface=row['surface'])


# 映射地质层序到表面
# 这一步将模型中的地质系列与实际的表面名称关联起来，
# 并告诉GemPy如何处理这些表面（例如，哪些是断层，哪些是地层）
gp.map_stack_to_surfaces(geo_model,
                        {"Fault_Series": ['Fault1', 'Fault2'],
                         "Stratigraphy_Series": ['Layer1', 'Layer2', 'Layer3', 'Layer4']},
                         )

# --- 4. 绘制3D模型 ---
# 确保所有数据都已加载和映射，现在可以进行3D绘图
print("尝试绘制3D模型...")
gpv.plot_3d(geo_model, plot_data=True) # plot_data=True 确保观测点被绘制出来