创建基于条件

我有一个数据帧创建基于条件

pd.DataFrame({"A":[0,1,0,1], 
      "B":[-1,0,0,0], 
      "C":[0,0,0,0]}, 
     index = [.1,.2,.3, .4])

我第一次在逻辑上接近问题的方法

for index, row in iterrows(): 
    if df['A'] == 1: 
     df['C'] == 1 
    elif df['B'] == -1 
     df['C'] == -1 
    else: 
     df['C'] == 0

我想

pd.DataFrame({"A":[0,1,0,1], 
      "B":[-1,0,0,0], 
      "C":[-1,1,0,1]}, 
     index = [.1,.2,.3, .4])

后的数据帧一列尝试第一种方法时，我尝试了其他问题中提出的各种方法，但没有一种方法适合我的问题。

来源

2017-10-11 Anonymous

你可以使用嵌套调用np.where：

df.C = np.where(df.A == 1, 1, np.where(df.B == -1, -1, 0)) 
df 
    A B C 
0.1 0 -1 -1 
0.2 1 0 1 
0.3 0 0 0 
0.4 1 0 1

性能

df = pd.concat([df] * 100000) 

%timeit np.select([df.A == 1, df.B == -1], [1, -1]) 
100 loops, best of 3: 5.25 ms per loop 

%timeit np.where(df.A == 1, 1, np.where(df.B == -1, -1, 0)) 
100 loops, best of 3: 2.86 ms per loop

来源

2017-10-11 01:06:05

使用numpy.select：

df['C'] = pd.np.select([df.A == 1, df.B == -1], [1, -1]) 

df 
#  A B C 
#0.1 0 -1 -1 
#0.2 1 0 1 
#0.3 0 0 0 
#0.4 1 -1 1

来源

2017-10-11 01:00:18 Psidom

创建基于条件

回答

相关问题