python - Pandas 进行字典查找的惯用方式

Question

我有一个 pandas 整数系列(它们仅限于一些小的有限子集)和这些可能的整数到 double 的字典。我想创建一个类似于 dictionary[series] 的新系列。 pandas 惯用的做法是什么？

Answer 1

您可以使用map来做到这一点。

例如，这是一个将一些整数 n 映射到 n + 0.5 的字典，以及一系列整数:

>>> d = {1: 1.5, 2: 2.5, 3: 3.5, 4: 4.5, 5: 5.5}
>>> s = pd.Series([1, 2, 3, 4, 5]
>>> s
0    1
1    2
2    3
3    4
4    5
dtype: int64

要创建新系列，您可以编写:

>>> s.map(d)
0    1.5
1    2.5
2    3.5
3    4.5
4    5.5
dtype: float64

除了字典之外，map 还可以接受另一个 Series 作为参数或函数。