Question

我想找到pandas系列中每个元素与0之间的成对最大值。我的粗略解如下：

import numpy as np
import pandas as pd
np.random.seed(1)

series = pd.Series(np.random.randn(100))
pmax = pd.Series([])
for i in range(len(series)):
    pmax[i] = max(series[i],0)

我需要在大量序列上运行此方法，并且此解决方案太慢。是否有矢量化的方法来达到相同的结果？

Answer 1

设置

s = pd.Series([1,2,3,-1,-2,3,4,-5])

使用mask作为填充值：

s.mask(s<0, 0)

0    1
1    2
2    3
3    0
4    0
5    3
6    4
7    0
dtype: int64

使用np.clip，没有上限：

np.clip(s, 0, None)

@Coldspeed建议使用pd.Series.clip_lower：

s.clip_lower(0)

时间

In [204]: %%timeit
     ...: pmax = pd.Series([])
     ...: for i in range(len(series)):
     ...:     pmax[i] = max(series[i],0)
     ...:
81.2 ms ± 4.06 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)

In [205]: %timeit series.mask(series<0, 0)
626 µs ± 30.6 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

In [206]: %timeit np.clip(series, 0, None)
124 µs ± 3.44 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)

In [209]: %timeit series.clip_lower(0)
97.2 µs ± 3.15 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)

Answer 2

我正在寻找针对Rs pmax()的python实现的解决方案，偶然发现了numpys maximum()函数的作用，而该函数正是pmax()所做的：

pmax(5,c(1,2,6)
[1] 5 5 6

并且：

>>> import numpy as np
>>> np.maximum(5, [1,2,6])
array([5, 5, 6])

熊猫系列成对最大

2 个答案: