简单地将整数值映射到另一个整数

Question

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

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

这是：假设我收到一个随机正整数作为输入，我需要将属于set1的那些映射到整数0，将set2的那些映射到{{1} }，以及2至set3中的内容。

在Python中是否有一种“简便”的方法？

Answer 1

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