Question

我的问题在以下视频链接中描述：

https://www.youtube.com/watch?v=nk5tBosK0iU

我不知道为什么我不能得到条件（df[condition]），其中condition是一个变量，适用于NaN值。

Answer 1

您需要pandas.isnull在自定义函数中检查标量NaN，但更快的解决方案是双numpy.where：

import pandas as pd
import numpy as np

df = pd.DataFrame({'CloseDelta':[np.nan,-0.5,0.5],
                   'B':[0,1,0]})

print (df)
  B  CloseDelta
0  0         NaN
1  1        -0.5
2  0         0.5

def f(x):
    if (pd.isnull(x)):
        return 0
    elif (x<0):
        return -1 
    else: 
        return 1

df['new'] = np.where(df.CloseDelta.isnull(), 0, np.where(df.CloseDelta<0, -1, 1))

df['new1'] = df.CloseDelta.apply(f)

print (df)
   B  CloseDelta  new  new1
0  0         NaN    0     0
1  1        -0.5   -1    -1
2  0         0.5    1     1

<强>计时：

#[300000 rows x 3 columns]
df = pd.concat([df]*100000).reset_index(drop=True)

In [28]: %timeit np.where(df.CloseDelta.isnull(), 0, np.where(df.CloseDelta<0, -1, 1))
100 loops, best of 3: 1.99 ms per loop

In [29]: %timeit df.CloseDelta.apply(f)
1 loop, best of 3: 245 ms per loop

如何根据Pandas中缺失值的另一列创建新列？

1 个答案: