euler_condicional.py

import os
import pandas as pd

NP = 103958
g = 9.81
L0 = 43.437
m = 65.832
k1 = (10/10000) * (NP - 100000) + 40
k2 = 0.80

def write_to_excel(rows):
    new_sheet = {}
    for i in range(len(rows[0])):
        column_name = rows[0][i]
        data = []
        for row in rows:
            data.append(row[i])

        new_sheet[column_name] = data

    new_df = pd.DataFrame(new_sheet)
    new_df = new_df.iloc[1:, :]

    filename = 'datos.xlsx'
    filePath = os.path.join(filename, filename)
    writer = pd.ExcelWriter(filePath, engine='xlsxwriter')

    new_df.to_excel(writer)

    writer.save()


def u_n1(un, vn, h):
    un1 = un + (h * vn)
    return un1


def v_n1(un, vn, h):
    if un >= L0:
        vn1 = vn + (h * (g - ((k1 * ((un - L0)**k2))/m)))
    else:
        vn1 = vn + (h * g)

    return vn1


def w_n1(un):
    if un >= L0:
        wn1 = (g - ((k1 * ((un - L0)**k2))/m))
    else:
        wn1 = g

    return wn1


def main():
    rows = [["N", "h", "un", "vn", "wn"]]
    un = 0
    vn = 0
    wn = 0

    h = 0.1
    i = 1000
    n = 0

    limit = 150 * 0.9
    cumple_1 = False
    cumple_2 = True
    cumple_3 = False

    u_max = 0
    a_max = 0

    rows.append([n, h, un, vn, wn])
    #print("n:", n, "| un:", un, "| vn:", vn, "| wn:", wn)

    while n < i:
        n += 1

        un1 = u_n1(un, vn, h)
        vn1 = v_n1(un1, vn, h)
        wn1 = w_n1(un)

        un = un1
        vn = vn1
        wn = wn1

        rows.append([n, h, un, vn, wn])

        if limit < un < 150:
            cumple_1 = True
        elif un > 150:
            cumple_2 = False

        if un > u_max:
            u_max = un
            a_max = wn

        #print("n:", n, "| un:", un, "| vn:", vn, "| wn:", wn)

    if abs(a_max) <= (2.5 * g):
        cumple_3 = True

    if cumple_1 and cumple_2 and cumple_3:
        print("Condiciones cumplidas! | k1:", k1, "| k2:", k2)
        print("U MAX:", u_max)
        print("A on MAX:", a_max)
    else:
        print("No cumple las condiciones | k1:", k1, "| k2:", k2)
        print("U MAX:", u_max)
        print("A on MAX:", a_max)

    write_to_excel(rows)


main()