NumPy中与Pandas Series.map等价的功能是什么？

Question

NumPy中与Pandas Series.map等价的功能是什么？

pythonpandasnumpy

4

我有一个查找值的字典 dictionary = {'a': 1, 'b': 2, 'c': 3, 'd': 3}

我想从numpy中得到的结果是，当传入字典时，pandas.Series.map()返回的结果。例如：series.map(dictionary, na_action='ignore')

注意：这个series.map()函数非常快，这让我相信在numpy API中一定有一个相应的函数，而不是我实现一些涉及numpy.where()和循环遍历字典键的解决方案。

- SARose

对于一般（非字典）情况，请参见在numpy数组上映射函数的最有效方法。 - jpp

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Divakar · Accepted Answer

这是一个NumPy的例子 -

def map_series_by_dict(s, d):
    a = s.values
    v = np.array(list(d.values()))
    k = np.array(list(d.keys()))    
    sidx = k.argsort()
    out_ar = v[sidx[np.searchsorted(k,a,sorter=sidx)]]
    return pd.Series(out_ar, index=s.index)

示例运行 -

In [143]: d
Out[143]: {'a': 1, 'b': 2, 'c': 3, 'd': 3}

In [144]: s
Out[144]: 
0    a
1    a
2    c
3    b
4    a
dtype: object

In [145]: map_series_by_dict(s, d)
Out[145]: 
0    1
1    1
2    3
3    2
4    1