整数值到另一个整数的简单映射

Question

这是一个非常简单的问题，但我还没有找到一个简单的方法来解决它。考虑以下整数集（它们不是实际的Pythonset，因为它们包含无限多的成员）：

set1 = (0, 3, 6, 9,  ...) --> 0
set2 = (1, 4, 7, 10, ...) --> 2
set3 = (2, 5, 8, 11, ...) --> 4

这是：假设我接收到一个随机的正整数作为输入，我需要将那些属于set1的映射到整数0，set2中的映射到2，set3中的映射到4。你知道吗

在Python中有“简单”的方法吗？你知道吗

Answer 1

def map_to_set(i):
    return 2*(i%3)