Python：公式作为矩阵

Question

我正在计算不同国家/地区的总运费。作为输入，我有一个初始费用和来源国家和目的地国家。作为输出，我想得到总成本。 当然，根据原产地和目的地国家，计算总成本的公式会有所不同。

为了解决这个问题，我想我应该使用一个矩阵，其中包含应用于计算总成本所需的公式。 我的问题是我真的不知道如何制作包含公式的矩阵。你能指出我正确的方向吗？

为了让它更容易理解，我在下面想到了矩阵的草图。

输入功能：

totalCostCalc(init_cost, origin_country, destination_country)

计算矩阵：

矩阵元素的不同颜色显示应用的不同公式或函数。

你可以通过展示如何开始帮助我，或者类似于我真正被卡住的东西。

Answer 1

我不知道你是否愿意接受任何其他解决方案，但是我会说，制作这样的公式矩阵并不是一件好事。

我认为你应该创建一个dictionary个作为key并且他们的value作为大陆。

因此，每当调用totalCostCalc函数时，您只需检查三个条件。

def totalCostCalc(init_cost, origin_country, origin_country_fee, destination_country, destination_country_fee)
    if (origin_country == destination_country):
        return init_cost
    elif (d[origin_country] == d[destination_country]):
        return init_cost*origin_country_fee*destination_country_fee
    else:
        return init_cost*origin_country_fee*destination_country_fee*continent_fee

其中d是定义为d = {'Germany': 'Europe', 'Hungary': 'Europe', 'China': 'Asia'}的字典。