优化应力计算

Signed-off-by: caner <5658514@qq.com>

main.py

@@ -0,0 +1,128 @@
+import pandas as pd
+from scipy.optimize import fsolve
+import glob
+import os
+import math
+
+# RCE方程函数
+def RCE(rce):
+    try:
+        RC = float(rce[0])
+        E = float(rce[1])
+        return [
+            2*K*E-RC*RC,
+            2*K1*E-RC*RC
+        ]
+    except:
+        return ['null', 'null']
+
+# R123方程函数
+def R123(r):
+    try:
+        R1 = float(r[0])
+        R2 = float(r[1])
+        R3 = float(r[2])
+        V = float(0.3)
+        return [
+            # 2*弹性模量（E）*机械比能-R1*R1-R2*R2-R3*R3+2*v*(R1*R2+R1*R3+R2*R3)
+            2*e*K-R1*R1-R2*R2-R3*R3+2*V*(R1*R2+R1*R3+R2*R3),
+            2*e1*K1-R1*R1-R2*R2-R3*R3+2*V*(R1*R2+R1*R3+R2*R3),
+            2*e2*K2-R1*R1-R2*R2-R3*R3+2*V*(R1*R2+R1*R3+R2*R3)
+        ]
+    except:
+        return ['null', 'null', 'null']
+
+# 对数据为null 的进行平均值填补
+def nullData(arr):
+    num = 0
+    lens = len(arr)
+    for j in arr:
+        if j != 'null':
+            num += j
+    ag = num/lens
+    for z in range(lens):
+        el = arr[z]
+        if el == 'null':
+            arr[z] = ag
+
+
+if __name__ == "__main__":
+    path = 'C:\\Users\\caner\\Desktop\\test'
+    colTitle = '机械比能'
+    try:
+        files = glob.glob(os.path.join(path, "*.xls"))
+        for file in files:
+            array = pd.read_excel(file) #读取excel数据
+            col = array[colTitle]  # 某列
+            colLen = len(col)  # 一列的长度
+            rowLen = len(array.iloc[1])  # 一行的长度
+            RC = [0]*(colLen)  # RC初始化
+            E = [0]*(colLen)  # E初始化
+            RK1 = [0]*(colLen)  # R1初始化
+            RK2 = [0]*(colLen)  # R2初始化
+            RK3 = [0]*(colLen)  # R3初始化
+            GRA = [0]*(colLen)  # 强度初始化
+
+            # 两行计算RCE
+            for i in range(colLen):
+                if (i+1) < len(col):
+                    K = col[i]
+                    K1 = col[i+1]
+                    res = fsolve(RCE, [100, 100000000000])  # 其它参数直接用
+                    rc = RCE(res)[0]
+                    e = RCE(res)[1]
+                    RC[i] = math.fabs(rc)  # 绝对值
+                    RC[i+1] = math.fabs(rc)  # 列最后一个值使用上一个的值
+                    E[i] = math.fabs(e)
+                    E[i+1] = math.fabs(e)
+
+            # 单独对数据为null的进行平均值计算
+            nullData(RC)
+            nullData(E)
+
+            # 三行计算R123
+            for j in range(colLen):
+                if (j+2) < len(col):
+                    K = col[j]
+                    K1 = col[j+1]
+                    K2 = col[j+2]
+                    e = E[j]
+                    e1 = E[j+1]
+                    e2 = E[j+2]
+                    rea = fsolve(R123, [1, 1, 1])  # 其它参数直接用
+                    rea = R123(rea)
+                    arr = sorted(rea, reverse=True) #倒叙
+                    RK1[j] = math.fabs(arr[0])
+                    RK1[j+1] = math.fabs(arr[0])
+                    RK1[j+2] = math.fabs(arr[0])
+                    RK2[j] = math.fabs(arr[1])
+                    RK2[j+1] = math.fabs(arr[1])
+                    RK2[j+2] = math.fabs(arr[1])
+                    RK3[j] = math.fabs(arr[2])
+                    RK3[j+1] = math.fabs(arr[2])
+                    RK3[j+2] = math.fabs(arr[2])
+
+            # 单独对数据为null的进行平均值计算
+            nullData(RK1)
+            nullData(RK2)
+            nullData(RK3)
+
+            # 计算强度
+            for z in range(colLen):
+                GRA[z] = RC[z] / RK1[z]
+
+            # 对整体数据进行倍数计算↓
+            # .......
+
+            # 保存excel
+            array.insert(rowLen, '单轴抗压强度', RC)
+            array.insert(rowLen+1, '弹性模量', E)
+            array.insert(rowLen+2, '第一主应力', RK1)
+            array.insert(rowLen+3, '第二主应力', RK2)
+            array.insert(rowLen+4, '第三主应力', RK3)
+            array.insert(rowLen+5, '岩石强度应力比', GRA)
+            pd.DataFrame(array).to_excel(file, sheet_name='Sheet1', index=False, header=True)
+            print('保存完毕')
+
+    except:
+        print('计算错误')