Реализация матрицы уравнений: использование цикла для определения функций (Python)

Question

Для контекста я по существу использую численный интегратор, который принимает набор дифференциальных уравнений, определяемых как функции. Большой набор этих функций соответствует регулярному шаблону, и я хотел бы определить их в цикле (или что-то наиболее подходящее). Например;

#system coordinates 
s = [y1,y2] 

#system equations 
def e1(s): 
    x1 = s[1]**2 + 1 
    return x1 

def e2(s): 
    x1 = s[2]**2 + 2 
    return x1 

#equations of motion 
eom = [e1,e2] 

Не все функции будет следовать точному шаблону, для тех, которые делают, хотя в идеале мне нужно что-то вроде,

def en(s) 
    x1 = s[n]**2 + n 
    return x1 

где можно перебрать диапазон значений «N» , Спасибо за любой совет.

Answer 1

Почему бы не просто использовать второй параметр в функции следующим образом:

def en(s, n) 
    x1 = s[n]**2 + n 
    return x1 

result = [] 
for i in range(100): # 100 is just for illustration purposes.. 
    result[0] = en(s, i) # you do not have to store them in a list. just an example 

Answer 2

Я хотел бы использовать partial, значения связывания Wich для аргументов функций:

import functools 

def e1(s, n, v1, v2): 
    x1 = s[n]**v1 + v2 
    return x1 

[functools.partial(e1, n=i, v1=2, v2=1) for i in range(10)] # this was your first example 

#your second example 
[functools.partial(e1, n=n, v1=2, v2=n) for n in range(10)]