迭代地根据字符串声明变量?

时间:2017-08-12 18:40:16

标签: python variables iteration

不确定之前是否曾询问过此问题。这是一个奇怪的问题,所以我会继续开火。

我有一些变量(或相当不变的)定义:

# Constants

# Colors
RED="RED"
ORANGE="ORANGE"
YELLOW="YELLOW"
GREEN="GREEN"
CYAN="CYAN"
BLUE="BLUE"
MAGENTA="MAGENTA"
# Modes
PANIC="PANIC"
SOLID="SOLID"
BREATHING="BREATHING"
# Special sub-modes (for panic)
BLINKING="BLINKING"

# Declare them
SOLID_RED="{}_{}".format(SOLID,RED)
SOLID_BLUE="{}_{}".format(SOLID,BLUE)
SOLID_MAGENTA="{}_{}".format(SOLID,MAGENTA)
## ..
BREATHING_RED="{}_{}".format(BREATHING,RED)
BREATHING_BLUE="{}_{}".format(BREATHING,BLUE)
BREATHING_MAGENTA="{}_{}".format(BREATHING,MAGENTA)
## ..
PANIC_RED="{}_{}".format(PANIC,RED)
PANIC_BLUE="{}_{}".format(PANIC,BLUE)
PANIC_MAGENTA="{}_{}".format(PANIC,MAGENTA)
## ..
PANIC_BLINKING="{}_{}".format(PANIC,BLINKING)

我有很多定义!不必像这样输入它们,我是否有办法只使用定义BEFORE # declare them或者使用字典来构造所有这些常量作为字符串存在?

这种迭代构造所需的格式是:MODE_COLOR命名约定。

我要求使用Python 2.7来解决这个问题。因为我有一些依赖的2.7 API。

3 个答案:

答案 0 :(得分:2)

我会使用字典作为存储变量的容器。只需列出列表中的所有颜色和模式,然后使用字典理解:

colors_list = ['red', 'blue']
modes_list = ['panic', 'solid']
color_modes = {k1 + '_' + k2: k1.upper() + '_' + k2.upper()
               for k1 in colors_list for k2 in modes_list}
>>> color_modes
{'blue_panic': 'BLUE_PANIC',
 'blue_solid': 'BLUE_SOLID',
 'red_panic': 'RED_PANIC',
 'red_solid': 'RED_SOLID'}

答案 1 :(得分:2)

使用itertools.combinationslocals()的另一种方式:

from itertools import combinations
from pprint import pprint

# Colors
RED="RED"
ORANGE="ORANGE"
YELLOW="YELLOW"
GREEN="GREEN"
CYAN="CYAN"
BLUE="BLUE"
MAGENTA="MAGENTA"
# Modes
PANIC="PANIC"
SOLID="SOLID"
BREATHING="BREATHING"
# Special sub-modes (for panic)
BLINKING="BLINKING"

v_consts = {k:v for k, v in locals().items() if k.isupper()}
combs = combinations(v_consts.values(), 2)
d_consts = {'%s_%s' % k: '%s_%s' % k for k  in combs}
pprint(d_consts)

# Edit:
# If you want to add the created variables in Python's scope
# You can do something like this
globals().update(d_consts)
print SOLID_BLINKING, type(SOLID_BLINKING)

输出:

{'BLINKING_CYAN': 'BLINKING_CYAN',
 'BLINKING_MAGENTA': 'BLINKING_MAGENTA',
 'BLINKING_ORANGE': 'BLINKING_ORANGE',
 'BLINKING_PANIC': 'BLINKING_PANIC',
 'BLINKING_RED': 'BLINKING_RED',
  ...
 'YELLOW_MAGENTA': 'YELLOW_MAGENTA',
 'YELLOW_ORANGE': 'YELLOW_ORANGE',
 'YELLOW_PANIC': 'YELLOW_PANIC',
 'YELLOW_RED': 'YELLOW_RED'}
 SOLID_BLINKING <type 'str'>

答案 2 :(得分:1)

我认为你要做的就是散发一些代码味道。

我可能采用的方法是使用字典和交叉产品。这是一个缩小的例子:

from itertools import product

A = ['a', 'b', 'c']
B = ['d', 'e', 'f']

AB = {"{0} {1}".format(a, b): "{0}_{1}".format(a, b) for a, b in product(A, B)}

print(AB)

您可以将其应用于颜色和修改器,并按名称访问颜色:

colors['Magenta Solid']