Как подставить массив в функцию?

Question

[Timebird]

Здравствуйте, вопрос в следующем:
имеется 4 переменные, содержащие 4 одномерных массива со значениями.
Нужно их подставить в функцию, тем самым получив на выходе один массив значений.
Для примера возьму такую функцию:
x(t) = a*sin(omega*t + phi)
Вот а, omega, phi и t -- это и есть 4 известных одномерных массива.

Можете подсказать, как синтаксически получить на выходе массив x?
Заранее спасибо!

Answer 1

[nirvimel]

def f(a, omega, phi, t):
    return a * sin(omega * t + phi)

x = [f(a[i], omega[i], phi[i], t[i]) for i in range(len(a))]

Comments

[Timebird]

Спасибо!
А скажите пожалуйста, почему для len задан параметр "а"?

[nirvimel]

Timebird: Если даны отдельные списки a, omega, phi, t, то они должны быть равной длины, если так, то все равно длину какого списка брать, пусть это будет, например a.

[Oscar Django]

Еще сахарку
x = [f(*z) for z in zip(a, omega, phi, t)]

[Oscar Django]

Или
x = map(f, a, omega, phi, t)

[Oscar Django]

Ато и вовсе

x = map(lambda a, omega, phi, t: a * sin(omega * t + phi), a, omega, phi, t)

[nirvimel]

Oscar Django: Считаю вариант с map наиболее красивым решением. Но вот подстановка лямбды - это решения только для очень простых функций. Если в будущем потребуется изменить (утяжелить, например) логику функции, то лучше бы она была записана отдельно от того места, где применяется, т.е. без лямбды.
Жалко комментарии нельзя плюсовать. Может оформите как ответ?

[Oscar Django]

Да легко

Answer 2

[Oscar Django]

По просьбам в дополнение к @nirvimel
Добавляем сахарок
x = [f(*z) for z in zip(a, omega, phi, t)]
И еще
x = map(f, a, omega, phi, t)
Для простых функций

x = map(lambda a, omega, phi, t: a * sin(omega * t + phi), a, omega, phi, t)