作为 Python 中键的集合/列表的成员身份

Question

例如，假设我有 n 集合，例如：

states1 = {"AL","AZ","CA"}
states2 = {"CO","NJ","NY"}
states3 = {"GA","IN","OK"}
...

集合是不相交的。

我有一个熊猫数据框，其中有一列“状态”。我想创建一个新列“level”，如果“state”在states1，则取值x，如果在states2，则取值y，依此类推，其中每个集合对应一个值。

有没有简单的方法来做到这一点？我能想出的最好方法是将这些集合声明为frozensets，并创建一个以这些集合为键的字典。然后我想我可以遍历这个字典的所有键并检查每个集合的状态成员资格并选择相应的值。

Answer 1

由于集合是不相交的，您可以像这样创建字典：

{"AL": x, "AZ": x, "CA": x, "CO": y, "NJ": y, "NY": y, ...}