別の列に基づくパンダデータフレームの列の作成

Question

パンダのデータフレームは1つの列に文字列のみが含まれています。

df= pd.DataFrame(
    { 
    "A": [2,4,7,17,39], 
    "B": ["apple","apple","broccoli","rose","apple"] 
    } 
) 

この列「B」を調べて、文字列に単語の特定の部分が含まれていることを確認します。次に、 "app"が行内にあるときはいつでも "fruit"、 "brocc"が出現するときはいつでも "ros"が表示され、野菜はいつでも表示される "flower"という一連の文字列を含む新しい列 "C"を作成します。

最終データフレームは、次のようになります。

df= pd.DataFrame(
    { 
    "A": [2,4,7,17,39], 
    "B": ["apple","apple","broccoli","rose","apple"], 
    "C": ["fruit","fruit", "vegetable", "flower", "fruit"] 
    } 
) 

Answer 1

あなたは、コンバータとしての辞書を使用することができ、ANそれはapplyのための入力としてget方法です：あなたが必要となる部分一致の場合

converter = {'apple': 'fruit', 
      'broccoli': 'veg', 
      'rose': 'flower'} 

df['C'] = df['B'].apply(converter.get) 
print(df) 
    A   B  C 
0 2  apple fruit 
1 4  apple fruit 
2 7 broccoli  veg 
3 17  rose flower 
4 39  apple fruit 

これを少し変更する：

converter = {'app': 'fruit', 
      'brocc': 'vegetable', 
      'ros': 'flower'} 

df['C'] = df['B'].apply(lambda original: next(val for key, val in converter.items() if key in original)) 
print(df) 
    A   B   C 
0 2  apple  fruit 
1 4  apple  fruit 
2 7 broccoli vegetable 
3 17  rose  flower 
4 39  apple  fruit 

next(val for key, val in converter.items() if key in original)は、「行」にある最初に見つかったキーの辞書から値を返します。

Answer 2

辞書

d = {'apple': 'fruit', 'broccoli': 'vegetable', 'rose': 'flower'} 

を作成し、それを使用mapかreplace（マップが速く、このためです）：

df['B'].map(d) 
Out: 
0  fruit 
1  fruit 
2 vegetable 
3  flower 
4  fruit 
Name: B, dtype: object 

df['B'].replace(d) 
Out: 
0  fruit 
1  fruit 
2 vegetable 
3  flower 
4  fruit 
Name: B, dtype: object