Python：根据for循环中的条件选择功能？

Question

抱歉，如果标题有些模糊。我将在这里更详细地解释一切。假设我有此代码：

def function1(k):
    return k * 2


def function2(k):
    return k ** 2


func = 'Square'

for i in range(1, 10):
    if func == 'Multiply':
        function1(i)
    elif func == 'Square':
        function2(i)

如何修改上面的代码，以便IF语句可以在循环外面进行？似乎没有必要检查每个迭代的func的值，因为它不会在内部发生变化。循环。我正在寻找的是这样的东西：

def function1(k):
    return k * 2


def function2(k):
    return k ** 2


func = 'Square'

if func == 'Multiply':
     f = function1()
elif func == 'Square':
     f = function2()

for i in range(1, 10):
    f(i)

如果某事还不够清楚，或者我要问的是不可能的。

Answer 1

将功能存储在dict中，然后您可以跳过语句。

def function1(k):
    return k * 2

def function2(k):
    return k ** 2

d = {"Multiply": function1, "Square": function2}

func = "Square"
f = d[func]
f(3)
# 9

Answer 2

您可以执行此操作：

funcs = {'Multiply':function1, 'Square':function2}
theFunc = funcs[func]
for i in range(1, 10):
    x = theFunc(i)
    print(x)

传递中的一个注：^操作员没有采用参数的正方形，而是在其上执行位XOR。