python: Simple lookup table