Excel 数据处理

博客地址:https://www.cnblogs.com/zylyehuo/
2023 年高教社杯全国大学生数学建模竞赛题目 -- B 题 多波束测线问题
图表格式

python 复制代码
import numpy as np
import matplotlib.pyplot as plt
from openpyxl import load_workbook
from matplotlib.ticker import FuncFormatter
from sklearn.linear_model import LinearRegression

wb = load_workbook("附件.xlsx")

x = []
y = []
z = []

# 创建一个3D图表
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

worksheet = wb.worksheets[0]

yy = -0.02
for line in worksheet.iter_rows(min_row=3):
    yy += 0.02
    xx = -0.02
    for ind in range(2, len(line)):
        xx += 0.02
        x.append(xx)
        y.append(yy)
        zz = line[ind].value
        z.append(zz / 1852)
        print(f"{line[ind].value:.2f},坐标为({xx:.2f},{yy:.2f})", end=' ')  # 索引值从 0 开始
    print()

# # 绘制散点图,x、y 和 z 作为坐标,小数点后两位精度
# 使用颜色映射根据 z 值着色
sc = ax.scatter(x, y, z, c=z, cmap='viridis', marker='o')
# 设置坐标轴标签
ax.set_xlabel('X-Label(MM)')
ax.set_ylabel('Y-Label(MM)')
ax.set_zlabel('Z-Value(NM)', labelpad=10)

# 设置图表标题
plt.title('Depth')


# 自定义刻度标签的显示格式
def format_func(value, tick_number):
    return f"{value:.2f}"


def format_func_z(value, tick_number):
    return f"{value:.5f}"


# 设置x、y、z轴的刻度标签格式
ax.xaxis.set_major_formatter(FuncFormatter(format_func))
ax.yaxis.set_major_formatter(FuncFormatter(format_func))
ax.zaxis.set_major_formatter(FuncFormatter(format_func_z))

# 添加颜色条
colorbar = plt.colorbar(sc, label='Z Value(NM)', pad=0.2)
colorbar.set_label('Z Value(NM)')

# 拟合切平面
# 创建一个 LinearRegression 模型
model = LinearRegression()

# 将 x 和 y 合并为一个特征矩阵
features = np.column_stack((x, y))

# 拟合模型
model.fit(features, z)

# 获取拟合的平面参数
coefficients = model.coef_
intercept = model.intercept_

a = coefficients[0]
b = coefficients[1]
c = 1.0

equation = f"z = {a:.5f} * x + {b:.5f} * y + {c:.5f}"
print(equation)

# 水平平面的法向量
horizontal_plane_normal = np.array([0, 0, 1])

# 计算两个法向量的点积
dot_product = np.dot([a, b, c], horizontal_plane_normal)

# 计算夹角(弧度)
angle_rad = np.arccos(dot_product / (np.linalg.norm([a, b, c]) * np.linalg.norm(horizontal_plane_normal)))

# 转换为角度
angle_deg = np.degrees(angle_rad)

print(f"夹角(弧度):{angle_rad:.2f}")
print(f"夹角(度):{angle_deg:.2f}")


# 定义一个函数来计算 z 值
def calculate_z(x_z, y_z):
    return coefficients[0] * x_z + coefficients[1] * y_z + intercept


# 绘制拟合的切平面
xx, yy = np.meshgrid(np.arange(min(x), max(x), 0.02), np.arange(min(y), max(y), 0.02))
zz = calculate_z(xx, yy)

# 绘制切平面
ax.plot_surface(xx, yy, zz, cmap='coolwarm', alpha=0.5)  # 使用 'coolwarm' 颜色映射,透明度0.5

# 显示图表
plt.show()
相关推荐
万粉变现经纪人2 分钟前
如何解决pip安装报错ModuleNotFoundError: No module named ‘python-dateutil’问题
开发语言·ide·python·pycharm·pandas·pip·httpx
跟橙姐学代码16 分钟前
Python 类的正确打开方式:从新手到进阶的第一步
前端·python·ipython
c8i17 分钟前
关于python中的变量中使用的下划线_总结
python
站大爷IP24 分钟前
用Flet打造跨平台文本编辑器:从零到一的Python实战指南
python
MC皮蛋侠客37 分钟前
使用python test测试http接口
开发语言·python·http
胡耀超1 小时前
5、Python-NumPy科学计算基础
开发语言·人工智能·python·深度学习·numpy
BIGSHU09232 小时前
java多线程场景3-并发处理和异步请求
java·开发语言·python
Source.Liu2 小时前
【Python自动化】 21.3 Pandas Series 核心数据结构完全指南
python·自动化·pandas
Sirius Wu2 小时前
私有化部署Ragflow的预训练模型
人工智能·python·语言模型·火山引擎
Mr_sun.3 小时前
Day04_苍穹外卖——套餐管理(实战)
开发语言·python