如何在Python中缩短if if和elif代码

Question

有两个变量， a和b ，每个变量可以是正数或负数。 需要根据四个条件设置标识符。 我怎样才能最小化下面给出的代码，以便可以实现相同的任务？ 列表理解或按位操作或类似的操作可能会缩短这一点，但我不知道如何做到这一点。

if a > 0 and b > 0:
    direction = 'NORTH EAST'

elif a > 0 and b < 0:
    direction = 'SOUTH EAST'

elif a < 0 and b < 0:
    direction = 'SOUTH WEST'

elif a < 0 and b > 0:
    direction = 'NORTH WEST'

Answer 1

您可以使用这样的条件表达式

("NORTH " if b > 0 else "SOUTH ") + ("EAST" if a > 0 else "WEST")

还有一个可以在这里使用的hack

["SOUTH ", "NORTH "][b > 0] + ["WEST", "EAST"][a > 0]

这是有效的，因为在Python中， 布尔值实际上是整数 。 Python中有以下内容

print 1 == True
# True
print 0 == False
# True

Answer 2

使用条件表达式：

direction = ' '.join(['NORTH' if a > 0 else 'SOUTH',
                      'EAST' if b > 0 else 'WEST'])

演示：

>>> for a in (-1, 1):
...     for b in (-1, 1):
...         print ' '.join(['NORTH' if a > 0 else 'SOUTH',
...                         'EAST' if b > 0 else 'WEST'])
... 
SOUTH WEST
SOUTH EAST
NORTH WEST
NORTH EAST

Answer 3

你看起来很好。 使用条件表达式是一回事：

>>> 
>>> a = 1
>>> b = -1
>>> lat = 'NORTH' if a > 0 else 'SOUTH'
>>> lon = 'EAST' if b > 0 else 'WEST'
>>> direction = '{} {}'.format(lat, lon)
>>> direction
'NORTH WEST'
>>>

Answer 4

或者使用字典作为地图

map[-1] = {-1:'SOUTH WEST', 1: 'NORTH WEST'}
map[1] = {-1:'SOUTH EST', 1: 'NORTH EST'}
direction = map[a/abs(a)][b/abs(b)]