Add files via upload

Author: JingShing

script/order_bot_program_module/dialogue_section/equation.py

@@ -0,0 +1,48 @@
+#change num
+from chinese_num_transform import chinese_in_string_transform
+#re
+import re
+#numpy
+import numpy as np
+
+def equation(self):
+    self.line_speaker('請講一下是甚麼題目讓你連一元一次方程式都不會解的？')
+    self.line_speaker('先講一下題目有幾個變數？')
+    # ex: 兩個 -> 2
+    var_num = int(chinese_in_string_transform(self.listener())[:-1])
+    self.line_speaker('有{}個變數，請說{}個方程式。'.format(var_num,var_num))
+    # AX=B
+    A = np.array([[0]*var_num for _ in range(var_num)])
+    B = np.array([[0] for _ in range(var_num)])
+    X = np.array([[None] for _ in range(var_num)])
+    for i in range(var_num):
+        math_line = self.listener()
+        math_line = re.sub(r'[負副富]+','-',math_line)
+        math_line = re.sub(r'等於','=',math_line)
+        math_line = re.sub(r'(?<=\D)(?=[xyz])|^(?=[xyz])','1',math_line)
+        math_line = chinese_in_string_transform(math_line)
+        if var_num >= 1:
+            finda = re.findall(r'([\+\-\d]+)x',math_line)
+            A[i][0] = int(finda[0]) if len(finda)>=1 else 0
+        if var_num >= 2:
+            finda = re.findall(r'x?([\+\-\d]+)y',math_line)
+            A[i][1] = int(finda[0]) if len(finda)>=1 else 0
+        if var_num >= 3:
+            finda = re.findall(r'x?y?([\+\-\d]+)z',math_line)
+            A[i][2] = int(finda[0]) if len(finda)>=1 else 0
+        findx = re.findall(r'=(.*)',math_line)
+        B[i][0] = int(findx[0]) if len(findx)>=1 else 0
+    have_ans=True
+    try:
+        X = np.linalg.inv(A).dot(B)
+    except:
+        have_ans=False
+    if have_ans:
+        if var_num==1:
+            self.line_speaker('X等於{}'.format(X[0][0]))
+        elif var_num==2:
+            self.line_speaker('X等於{}\nY等於{}'.format(X[0][0],X[1][0]))
+        else:
+            self.line_speaker('X等於{}\nY等於{}Z等於{}'.format(X[0][0],X[1][0],X[2][0]))
+    else:
+        self.line_speaker('此題無解^_^')