python - 根据另一列在 Pandas 数据框中创建一列

Question

我有一个 pandas 数据框，其中一列仅包含字符串。

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

我想检查“B”列并查找字符串中包含单词特定部分的所有时间。然后，我创建了一个新列“C”，其中包含一系列字符串，只要“app”在行中，就会显示“fruit”，每当“ros”出现时，就会显示“flower”，只要出现“brocc”，就会显示 vegetable。

最终数据框将如下所示:

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

Answer 1

创建字典

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

并在 map 或 replace 中使用它(map 对此更快):

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