如何基於另一個在熊貓中缺失值的列創建新列？

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