Python一键生成井身结构图

Python一键生成井身结构图1.准备如下.CSV文件井眼轨迹数据及固井数据，如水泥返高等。

如下图2.将程序里面的读取路径更改为上述CSV文件存放路径。

3.运行程序，即可生成如下井身结构图。

4.Python代码如下：import numpy as npimport matplotlib.pyplot as pltimport pandas as pdimport cv2from PIL import Image, ImageDraw, ImageFontdata = pd.read_csv(r'C:\Users\Administrator\Desktop\2.CSV',encoding='gbk')#编码中文# 获取“井眼垂深”，“水平位移”，存放于数组data1 = data['TVD depth'] # 获取名字为TVD depth-列的数据coordinate_H = data1.values.tolist() # 将csv文件中TVD depth列中的数据保存到列表中data1 = data['Closure Distance'] # 获取名字为Closure Distance列的数据coordinate_W = data1.values.tolist() # 将csv文件中Closure Distance列中的数据保存到列表中data1 = data['key_points'] # 获取名字为key_points列的数据key_points = data1.values.tolist() # 将csv文件中key_points列中的数据保存到列表中# 平移函数def pingyi(list, s):list1 = []for i in list:list1.append(i + s)list = list1return list# width = 0.35 # the width of the bars: can also be len(x) sequence# fig, ax = plt.subplots()# ax.plot(coordinate_W, coordinate_H, color='tab:blue')plt.xticks([]) # 去掉x轴plt.yticks([]) # 去掉y轴plt.axis('off') # 去掉坐标轴plt.xlim(-5000, 7000) # 设置X轴范围plt.ylim(-800, 3500) # 设置Y轴范围# 设置plt.plot（）划线的线条属性color = (0, 0, 0)linewidth = 1linestyle = '-'#设置文本属性fontsize = 8left = 260right = 0# 初始化向左或向右平移距离s = [-420, 420, -300, 300, -210, 210, -140, 140]# s1 = 350#向左或者向右平移距离# s2 = -350#向左或者向右平移距离# s3 = 300#向左或者向右平移距离# s4 = -300#向左或者向右平移距离# 自定义井口数据a = -100 # 井口垂直增加长度b = 1 # 井口垂直增加长度,起点坐标上移1，避免重复划线# x0 = [coordinate_W[0],coordinate_W[0]+b]#垂直线# y0 = [coordinate_H[0],a+b]#垂直线# plt.plot(x0, y0, color=(0,0,0),linewidth=1, linestyle='-')# plt.plot(coordinate_W, coordinate_H, color=(0,0,0),linewidth=1, linestyle='-')h = 0 # 标记一开、二开及三开等下深位置（垂深）# 中间线middle_point_x = [coordinate_W[h], coordinate_W[h]]middle_point_y = [coordinate_H[h] + b - 350, coordinate_H[h] - 40]plt.plot(middle_point_x, middle_point_y, color='red', linewidth=1,linestyle='-.')plt.plot(coordinate_W, coordinate_H, color='red', linewidth=1, linestyle='-.')#海平面，泥面#左边l = -560r = 560plt.text(coordinate_W[h]-1150+l, coordinate_H[h]-80,r'~~~~~',fontsize=8,color='black')plt.text(coordinate_W[h]-1165+l, coordinate_H[h]-40,r'~~~~~',fontsize=8,color='black')plt.text(coordinate_W[h]-1150+l, coordinate_H[h]-0,r'~~~~~',fontsize=8,color='black')# plt.text(coordinate_W[h]-1165+l, coordinate_H[h]+40,r'~~~~~',fontsize=8,color='black')plt.text(coordinate_W[h]-1150+l, coordinate_H[h]+130,r'############',fontsize=4,color='black')plt.text(coordinate_W[h]-1150+l, coordinate_H[h]+165,r'############',fontsize=4,color='black')# plt.text(coordinate_W[h]-1150+l, coordinate_H[h]+150,r'############',fontsize=4,color='black')#右边plt.text(coordinate_W[h]+680+r, coordinate_H[h]-80,r'~~~~~',fontsize=8,color='black')plt.text(coordinate_W[h]+695+r, coordinate_H[h]-40,r'~~~~~',fontsize=8,color='black')plt.text(coordinate_W[h]+680+r, coordinate_H[h]-0,r'~~~~~',fontsize=8,color='black')# plt.text(coordinate_W[h]+695+r, coordinate_H[h]+40,r'~~~~~',fontsize=8,color='black')plt.text(coordinate_W[h]+680+r, coordinate_H[h]+130,r'############',fontsize=4,color='black')plt.text(coordinate_W[h]+680+r, coordinate_H[h]+165,r'############',fontsize=4,color='black')# plt.text(coordinate_W[h]+680+r, coordinate_H[h]+150,r'############',fontsize=4,color='black')"""i = 0,1,2,3,4,5,6,7。

标记向左或向右位置"""# 13-3/8"套管x = [s[0], s[0] + b] # 垂直线y = [coordinate_H[h], a + b] # 垂直线plt.plot(x, y, color=color, linewidth=linewidth, linestyle=linestyle) # 井口x = [s[1], s[1] + b] # 垂直线y = [coordinate_H[h], a + b] # 垂直线plt.plot(x, y, color=color, linewidth=linewidth, linestyle=linestyle) # 井口# 隔水管# 左边hh = 150#上下调隔水管高度p = 0while p < 60:x = [s[0] - 10 - 1.8 * p, s[0] - 10 - 1.8 * p]y = [coordinate_H[h] + hh + p, coordinate_H[h] + hh + 60]p = p + 0.1plt.plot(x, y, color=color, linewidth=0.1, linestyle=linestyle)# 右边p = 0while p < 60:x = [s[1] + 16 + 1.8 * p, s[1] + 16 + 1.8 * p]y = [coordinate_H[h] + hh + p, coordinate_H[h] + hh + 60]p = p + 0.1plt.plot(x, y, color=color, linewidth=0.1, linestyle=linestyle)""" 中文注释"""plt.text(s[0] - 10 - 1.8 * 60 - 1500 + left, coordinate_H[h] + hh + 60 + 80,r'24"隔水管：144.00m', fontsize=fontsize,color='black')# 9-5/8"套管x = [s[2], s[2] + b] # 垂直线y = [coordinate_H[h], a + b] # 垂直线plt.plot(x, y, color=color, linewidth=linewidth, linestyle=linestyle) # 井口x = [s[3], s[3] + b] # 垂直线y = [coordinate_H[h], a + b] # 垂直线plt.plot(x, y, color=color, linewidth=linewidth, linestyle=linestyle) # 井口# 7"套管x = [s[4], s[4] + b] # 垂直线y = [coordinate_H[h], a + b] # 垂直线plt.plot(x, y, color=color, linewidth=linewidth, linestyle=linestyle) # 井口x = [s[5], s[5] + b] # 垂直线y = [coordinate_H[h], a + b] # 垂直线plt.plot(x, y, color=color, linewidth=linewidth, linestyle=linestyle) # 井口"""# 7"套管x = [s[6], s[6] + b] # 垂直线y = [coordinate_H[h], a + b] # 垂直线plt.plot(x, y, color=color, linewidth=linewidth, linestyle=linestyle) # 井口x = [s[7], s[7] + b] # 垂直线y = [coordinate_H[h], a + b] # 垂直线plt.plot(x, y, color=color, linewidth=linewidth, linestyle=linestyle) # 井口"""# 对称双水平线longline1 = 1000 # 控制线长m1 = [coordinate_W[h] + s[4] + 65, coordinate_W[h] + s[4] + 65 - longline1] n1 = [coordinate_H[h] + a, coordinate_H[h] + a]plt.plot(m1, n1, color=(0, 0, 0), linewidth=3, linestyle='-')m2 = [coordinate_W[h] - s[4] - 65, coordinate_W[h] - s[4] - 65 + longline1] n2 = [coordinate_H[h] + a, coordinate_H[h] + a]plt.plot(m2, n2, color=(0, 0, 0), linewidth=3, linestyle='-')# 井口长水平线m3 = [coordinate_W[h] + s[4] + 65 - longline1, coordinate_W[h] - s[4] - 65 + longline1]n3 = [coordinate_H[h] + a - 80, coordinate_H[h] + a - 80]plt.plot(m3, n3, color=(0, 0, 0), linewidth=3, linestyle='-')"""中文-转盘面，泥面等"""r1 = 650plt.text(coordinate_W[h] - s[4] - 65 + longline1+r1, coordinate_H[h] + a - 80+40, r'转盘面：0.00m',fontsize=fontsize,color='black')plt.text(coordinate_W[h] - s[4] - 65 + longline1+r1, coordinate_H[h] + a - 80+40+100, r'海平面：58.00m',fontsize=fontsize,color='black')plt.text(coordinate_W[h] - s[4] - 65 + longline1+r1, coordinate_H[h] + a - 80+40+200, r'水深：36.00m',fontsize=fontsize,color='black')plt.text(coordinate_W[h] - s[4] - 65 + longline1+r1, coordinate_H[h] + a - 80+40+300, r'泥面：94.00m',fontsize=fontsize,color='black')# 井口短水平线m4 = [coordinate_W[h] + s[4] + 25, coordinate_W[h] - s[4] - 25]n4 = [coordinate_H[h] + a - 80 - 55, coordinate_H[h] + a - 80 - 55]plt.plot(m4, n4, color=(0, 0, 0), linewidth=3, linestyle='-')# 井口双短垂直线m5 = [coordinate_W[h] + s[4] + 25, coordinate_W[h] + s[4] + 25]n5 = [coordinate_H[h] + a - 80 - 55, coordinate_H[h] + a - 80]plt.plot(m5, n5, color=(0, 0, 0), linewidth=3, linestyle='-')m6 = [coordinate_W[h] - s[4] - 25, coordinate_W[h] - s[4] - 25]n6 = [coordinate_H[h] + a - 80 - 55, coordinate_H[h] + a - 80]plt.plot(m6, n6, color=(0, 0, 0), linewidth=3, linestyle='-')k = 7#控制绘制水平线时与垂直线不重复#z1控制上水平线上移；z2控制下水平线下移z1 = 4z2 = 4while h < len(coordinate_H):"""# 13-3/8"套管"""#套管鞋if coordinate_H[h] == key_points[0]:h1 = h#左边--双水平线m7 = [pingyi([coordinate_W[h]], s[0]+k), pingyi([coordinate_W[h]],s[2]-k)]n7 = [coordinate_H[h]+z2, coordinate_H[h]+z2]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle) # 右边--双水平线m7 = [pingyi([coordinate_W[h]], s[1] - k), pingyi([coordinate_W[h]],s[3] + k)]n7 = [coordinate_H[h]+z2, coordinate_H[h]+z2]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle)"""中文注释"""plt.text(coordinate_W[h]-2150+left, coordinate_H[h]+z2-50, r'13-3/8"套管：1800m',fontsize=fontsize,color='black')plt.text(coordinate_W[h] - 2150+left, coordinate_H[h] + z2 + 100, r'16"井眼：1805m', fontsize=fontsize, color='black')# 套管鞋底部水泥返高if coordinate_H[h] == key_points[1]:h2 = h#左边m7 = [pingyi([coordinate_W[h]], s[0]+k), pingyi([coordinate_W[h]],s[2]-k)]n7 = [coordinate_H[h]-z1, coordinate_H[h]-z1]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle) # 右边m7 = [pingyi([coordinate_W[h]], s[1] - k), pingyi([coordinate_W[h]],s[3] + k)]n7 = [coordinate_H[h]-z1, coordinate_H[h]-z1]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle)"""中文注释"""plt.text(coordinate_W[h] + 670+right, coordinate_H[h] + z2 + 200, r'13-3/8"套管固井尾浆：1400-1800m', fontsize=fontsize,color='black')# 套管顶部水泥返高（上）if coordinate_H[h] == key_points[2]:h3 = h# 左边m7 = [pingyi([coordinate_W[h]], s[0] + k), pingyi([coordinate_W[h]],s[2] - k)]n7 = [coordinate_H[h]-z1, coordinate_H[h]-z1]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle) # 右边m7 = [pingyi([coordinate_W[h]], s[1] - k), pingyi([coordinate_W[h]],s[3] + k)]n7 = [coordinate_H[h]-z1, coordinate_H[h]-z1]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle)"""中文注释"""plt.text(coordinate_W[h] + 540+right, coordinate_H[h] + z2 + 200, r'13-3/8"套管固井首浆：150-600m', fontsize=fontsize, color='black')# 套管顶部水泥返高（下）if coordinate_H[h] == key_points[3]:h4 = h# 左边m7 = [pingyi([coordinate_W[h]], s[0] + k), pingyi([coordinate_W[h]],s[2] - k)]n7 = [coordinate_H[h]+z2, coordinate_H[h]+z2]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle) # 右边m7 = [pingyi([coordinate_W[h]], s[1] - k), pingyi([coordinate_W[h]],s[3] + k)]n7 = [coordinate_H[h]+z2, coordinate_H[h]+z2]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle)"""# 9-5/8"套管"""# 套管鞋if coordinate_H[h] == key_points[4]:h5 = h# 左边--双水平线m7 = [pingyi([coordinate_W[h]], s[2] + k), pingyi([coordinate_W[h]],s[4] - k)]n7 = [coordinate_H[h]+z2, coordinate_H[h]+z2]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle) # 右边--双水平线m7 = [pingyi([coordinate_W[h]], s[3] - k), pingyi([coordinate_W[h]],s[5] + k)]n7 = [coordinate_H[h]+z2, coordinate_H[h]+z2]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle)"""中文注释"""plt.text(coordinate_W[h] - 1950+left, coordinate_H[h] + z2 - 50, r'9-5/8"套管：3355m', fontsize=fontsize, color='black')plt.text(coordinate_W[h] - 1950+left, coordinate_H[h] + z2 + 100, r'12-1/4"井眼：3360m', fontsize=fontsize, color='black')# 套管鞋底部水泥返高if coordinate_H[h] == key_points[5]:h6 = h# 左边m7 = [pingyi([coordinate_W[h]], s[2] + k), pingyi([coordinate_W[h]],s[4] - k)]n7 = [coordinate_H[h]-z1, coordinate_H[h]-z1]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle) # 右边m7 = [pingyi([coordinate_W[h]], s[3] - k), pingyi([coordinate_W[h]],s[5] + k)]n7 = [coordinate_H[h]-z1, coordinate_H[h]-z1]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle)"""中文注释"""plt.text(coordinate_W[h] + 570+right, coordinate_H[h] + z2 + 200, r'9-5/8"套管固井尾浆：2950-3350m', fontsize=fontsize,color='black')# 套管顶部水泥返高（上）if coordinate_H[h] == key_points[6]:h7 = h# 左边m7 = [pingyi([coordinate_W[h]], s[2] + k), pingyi([coordinate_W[h]],s[4] - k)]n7 = [coordinate_H[h]-z1, coordinate_H[h]-z1]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle) # 右边m7 = [pingyi([coordinate_W[h]], s[3] - k), pingyi([coordinate_W[h]],s[5] + k)]n7 = [coordinate_H[h]-z1, coordinate_H[h]-z1]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle)"""中文注释"""plt.text(coordinate_W[h] + 600+right, coordinate_H[h] + z2 + 200, r'9-5/8"套管固井首浆：1650-2050m', fontsize=fontsize,color='black')# 套管顶部水泥返高（下）if coordinate_H[h] == key_points[7]:h8 = h# 左边m7 = [pingyi([coordinate_W[h]], s[2] + k), pingyi([coordinate_W[h]],s[4] - k)]n7 = [coordinate_H[h]+z2, coordinate_H[h]+z2]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle) # 右边m7 = [pingyi([coordinate_W[h]], s[3] - k), pingyi([coordinate_W[h]],s[5] + k)]n7 = [coordinate_H[h]+z2, coordinate_H[h]+z2]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle)"""# 7"套管"""# 套管鞋if coordinate_H[h] == key_points[8]:h9 = h# 左边--双水平线m7 = [pingyi([coordinate_W[h]], s[4] + k), pingyi([coordinate_W[h]],s[5] - k)]# n7 = [coordinate_H[h]+z2, coordinate_H[h]+z2]# plt.plot(m7, n7, color=color, linewidth=linewidth,linestyle=linestyle)# # 右边--双水平线# m7 = [pingyi([coordinate_W[h]], s[5] - k), pingyi([coordinate_W[h]], s[7] + k)]n7 = [coordinate_H[h]+z2, coordinate_H[h]+z2]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle)"""中文注释"""plt.text(coordinate_W[h]-1800+left, coordinate_H[h] + z2 - 50, r'7"套管：3883m', fontsize=fontsize, color='black')plt.text(coordinate_W[h]-1800+left, coordinate_H[h] + z2 + 100, r'8-1/2"井眼：3888m', fontsize=fontsize, color='black')# 套管顶部水泥返高if coordinate_H[h] == key_points[9]:h10 = h# 左边m7 = [pingyi([coordinate_W[h]], s[4] + k), pingyi([coordinate_W[h]],s[6] - k)]n7 = [coordinate_H[h]-z1, coordinate_H[h]-z1]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle) # 右边m7 = [pingyi([coordinate_W[h]], s[5] - k), pingyi([coordinate_W[h]],s[7] + k)]n7 = [coordinate_H[h]-z1, coordinate_H[h]-z1]plt.plot(m7, n7, color=color, linewidth=linewidth, linestyle=linestyle)"""中文注释"""plt.text(coordinate_W[h] + 570+right, coordinate_H[h] + z2 + 200, r'7"套管固井首浆：-3200m', fontsize=fontsize,color='black')plt.text(coordinate_W[h] + 570+right, coordinate_H[h] + z2 + 350, r'7"套管固井尾浆：返至油层顶以上150m', fontsize=fontsize,color='black')h = h + 1"""各层套管井身""""""# 13-3/8"套管"""# 左边--套管plt.plot(pingyi(coordinate_W[0:h1 + 1], s[0]), coordinate_H[0:h1 + 1],color=color, linewidth=linewidth,linestyle=linestyle)# 右边--套管plt.plot(pingyi(coordinate_W[0:h1 + 1], s[1]), coordinate_H[0:h1 + 1],color=color, linewidth=linewidth,linestyle=linestyle)"""# 9-5/8"套管"""# 左边--套管plt.plot(pingyi(coordinate_W[0:h5+1], s[2]), coordinate_H[0:h5+1], color=color, linewidth=linewidth, linestyle=linestyle)# 右边--套管plt.plot(pingyi(coordinate_W[0:h5+1], s[3]), coordinate_H[0:h5+1], color=color, linewidth=linewidth, linestyle=linestyle)"""# 7"套管"""# 左边--套管plt.plot(pingyi(coordinate_W, s[4]), coordinate_H, color=color,linewidth=linewidth, linestyle=linestyle)# 右边--套管plt.plot(pingyi(coordinate_W, s[5]), coordinate_H, color=color,linewidth=linewidth, linestyle=linestyle)"""# 7"套管"""# 左边--套管plt.plot(pingyi(coordinate_W[h10:], s[6]), coordinate_H[h10:], color=color, linewidth=linewidth, linestyle=linestyle)# 右边--套管plt.plot(pingyi(coordinate_W[h10:], s[7]), coordinate_H[h10:], color=color, linewidth=linewidth, linestyle=linestyle)"""颜色填充"""color1 = (0.4,0.4,0.4)linewidth1 = 0.1linestyle1 = '-'# 13-3/8"套管k1 = 16while k1 < 120-16:"""套管鞋"""# 左边--套管plt.plot(pingyi(coordinate_W[h2:h1 + 1], s[0]+k1), coordinate_H[h2:h1 + 1], color=color1, linewidth=linewidth1,linestyle=linestyle1)# 右边--套管plt.plot(pingyi(coordinate_W[h2:h1 + 1], s[1]-k1), coordinate_H[h2:h1 + 1], color=color1, linewidth=linewidth1,linestyle=linestyle1)"""套管顶"""# 左边--套管plt.plot(pingyi(coordinate_W[h3:h4 + 1], s[0] + k1), coordinate_H[h3:h4 + 1], color=color1, linewidth=linewidth1,linestyle=linestyle1)# 右边--套管plt.plot(pingyi(coordinate_W[h3:h4 + 1], s[1] - k1), coordinate_H[h3:h4 + 1], color=color1, linewidth=linewidth1,linestyle=linestyle1)k1 = k1 + 0.1# 9-5/8"套管k2 = 16while k2 < 90-16:"""套管鞋"""# 左边--套管plt.plot(pingyi(coordinate_W[h6:h5 + 1], s[2]+k2), coordinate_H[h6:h5 + 1],color=color1, linewidth=linewidth1,linestyle=linestyle1)# 右边--套管plt.plot(pingyi(coordinate_W[h6:h5 + 1], s[3]-k2), coordinate_H[h6:h5 + 1], color=color1, linewidth=linewidth1,linestyle=linestyle1)"""套管顶"""plt.plot(pingyi(coordinate_W[h7:h8+1], s[2]+k2), coordinate_H[h7:h8+1], color=color1, linewidth=linewidth1,linestyle=linestyle1)# 右边--套管plt.plot(pingyi(coordinate_W[h7:h8+1], s[3]-k2), coordinate_H[h7:h8+1], color=color1, linewidth=linewidth1,linestyle=linestyle1)k2 = k2 + 0.1# 7"套管k3 = 16while k3 < 70-16:# 左边--套管plt.plot(pingyi(coordinate_W[h10:], s[4]+k3), coordinate_H[h10:],color=color1, linewidth=linewidth1, linestyle=linestyle1)# 右边--套管plt.plot(pingyi(coordinate_W[h10:], s[5]-k3), coordinate_H[h10:],color=color1, linewidth=linewidth1, linestyle=linestyle1)k3 = k3 + 0.1# 符号和线型之间的对应关系# - 实线# -- 短线# -. 短点相间线# ：虚点线ax = plt.gca() # 获取到当前坐标轴信息ax.xaxis.set_ticks_position('top') # 将X坐标轴移到上面ax.invert_yaxis() # 反转Y坐标轴plt.rcParams['font.sans-serif'] = ['Microsoft YaHei']"""黑体/SimHei,仿宋/FangSong,楷体/KaiTi,微软雅黑体/Microsoft YaHei,宋体/SimSun, 隶书/LiSu,幼圆/YouYuan,华文细黑/STXihei,华文楷体/STKaiti,华文宋体,STSong,华文中宋/STZhongsong,华文仿宋/STFangsong,方正舒体/FZShuTi,方正姚体/FZYaoti, 华文彩云/STCaiyun,华文琥珀,STHupo,华文隶书/STLiti,华文行楷/STXingkai,华文新魏/STXinwei"""plt.show()。