physics-labs-collection/main.py

54 lines
1.9 KiB
Python

import os, sys
def avg(a):
return sum(a) / len(a)
def process_table(file):
table = [["L2", "L1", "t2", "m", "Δm", "t2i", "v1", "(v1)^2", "a", "ai", "❬a❭", "A", "B"]]
supported_args = ["r"]
config = {"r": 6}
with open(file, "r", encoding = "UTF-8") as f:
for i in f:
if i.rstrip("\r\n") != "":
if i.rstrip("\r\n")[0] == "#" or i.rstrip("\r\n")[0] == ";":
print(f"Special line: {i}")
try:
cline = i.rstrip("\r\n")[1:].split()
if len(cline) == 2 and cline[0] in supported_args:
config[cline[0]] = int(cline[1])
except:
pass
else:
print(f"Table line: {i}")
table.append(list(map(lambda x: float(x), i.rstrip("\r\n").replace(",", ".").split())))
r = config["r"]
for i in table[1:]:
i.append(round(i[0] / i[2], r))
i.append(round(i[6]**2, r))
i.append(round((i[0]**2)/(2*(i[2]**2)*i[1]), r))
preset_L2 = table[1][0]
preset_L1 = table[1][1]
for i in table[1:]:
i.append(round((preset_L2**2) / (2 * (i[5]**2) * preset_L1), r))
#i.append(round((i[0]**2) / (2 * (i[5]**2) * i[1]), r))
preset_avg_a = [avg(list(zip(*table[1:][i:i+3]))[9]) for i in range(0, 12, 3)]
for i in range(4):
for k in range(3):
table[1 + i*3 + k].append(round(preset_avg_a[i], r))
for i in table[1:]:
i.append(round(i[4]/(2*2*i[3]), r))
i.append(round(i[10]/(9.8-i[10]), r))
w = max([max(list(map(lambda x: len(str(x)), i))) for i in table])
print("\n".join([" | ".join(list(map(lambda x: str(x).center(w), i))) for i in table]))
if len(sys.argv) == 2 and os.path.exists(sys.argv[1]):
process_table(sys.argv[1])
else:
print("Будь ласка, вкажіть файл для обробки")