既存の列のbool値に基づいてpandasデータフレームに列を追加する

Question

新しい列を追加したいpandasデータフレームがあります。新しい列の値は、boolを含むデータフレーム内の既存の列によって決まります。以下のコードはPythonで適用されたC++のロジックですが、これを行うにはもっと「pythonic」な方法が必要です。 'isfixed'はboolsが含まれており、新しい列が助けを事前に'color code'

for i in range(data_2015['isfixed'].count()): 
    if data_2015['isfixed'][i] == True: 
     data_2015['color code'][i] = 'Blue' 
    else: 
     data_2015['color code'][i] = 'Green' 

感謝されます！

Answer 1

あなたはnumpy.whereを使用することができます。

import numpy as np 
data_2015['color_code'] = np.where(data_2015['isfixed'], 'Blue', 'Green') 

デモ：純粋なパンダのソリューションについては

df = pd.DataFrame({'isfixed': [True, False, True]}) 

df 
Out: 
    isfixed 
0 True 
1 False 
2 True 


df['color_code'] = np.where(df['isfixed'], 'Blue', 'Green') 

df 
Out: 
    isfixed color_code 
0 True  Blue 
1 False  Green 
2 True  Blue 

Answer 2

：

df = pd.DataFrame({'isfixed': [True, False, True]})

df['color_code'] = ["Blue" if value==True else "Green" for value in df['isfixed']]