Как применить функцию к нескольким переменным

Question

Как применить функцию к нескольким переменным

1

Мне нужно проверить несколько переменных и вызвать на них одну и ту же функцию

if isinstance(VarA,complex):
    VarA=SomeFunction(VarA)
if isinstance(VarB,complex):
    VarB=SomeFunction(VarB)
if isinstance(VarC,complex):
    VarC=SomeFunction(VarC)

Можно ли написать этот код короче/читабельнее?

hokisajam 21 июнь 2018, в 13:40

Источник

Теги:

python

python-3.x

functional-programming

3 ответа

2

Одним из решений является использование словаря для хранения вашего переменного количества переменных.

Затем выполните итерацию словаря и измените значения, если ваше условие выполнено.

vars = {'A': VarA, 'B': VarB, 'C': VarC}

for k, v in vars.items():
    if isinstance(v, complex):
        vars[k] = SomeFunction(vars[k])

jpp 21 июнь 2018, в 10:12

1

Я выбираю другой ответ, потому что у меня небольшое количество переменных, но ваш ответ тоже хорош.
hokisajam 21 июнь 2018, в 13:28

0

Вы можете сделать так,

varA,varB,varC = (SomeFunction(var) if isinstance(var,complex) else var for var in [varA,varB,varC])

Rahul K P 21 июнь 2018, в 09:54

Ещё вопросы

Я выбираю другой ответ, потому что у меня небольшое количество переменных, но ваш ответ тоже хорош.

ForceBru · Accepted Answer · 2018-06-21T08-32-00.000Z

Вы можете сделать это:

a, b, c = (f(x) if isinstance(x, complex) else x for x in [a, b, c])

Это становится более уродливым, когда имеется более трех переменных, поэтому, если вы нуждаетесь в большом количестве переменных, которые используются для выполнения одного и того же вычисления, вы должны подумать об использовании списков и циклов:

List = [1, 2, 3 + 1j, 4, 5 - 2j]

processed = [f(x) if isinstance(x, complex) else x for x in List]

Хорошее решение для небольшого количества предметов.