时间戳

首页 > 解决方案 > 如何根据类或组将变量分配给函数？

python - 如何根据类或组将变量分配给函数？

问题描述

我想做的是这样的：

if (color = "red"):
    a = 1
    b = 2
    c = 3
    break

if (color = "blue"):
    a = 5
    b = 6
    c = 7
    break

color = "blue"
z = a + b + c


print("The answer is = " + str(z))

我有几个类想要分配变量的值，然后根据我定义的类提取这些值。这可能吗？

标签： python

解决方案

假设您将类作为一个组而不是编程定义，您可以将字典用于可变数量的变量。

在这种情况下，您可以使用嵌套字典并以 O(1) 复杂度访问值。

d = {'red': {'a': 1, 'b': 2, 'c': 3},
     'blue': {'a': 5, 'b': 6, 'c': 7}}

color = 'blue'
d_color = d[color]
z = d_color['a'] + d_color['b'] + d_color['c']

print("The answer is = " + str(z))
# The answer is = 18

sum或者，更简单地说，您可以通过and对字典中的所有值求和dict.values：

z = sum(d['blue'].values())

推荐阅读