现存在df1,存在A列记录客户,B列记录供应商,现新建C列判断,若A列存在"信达"字符串,则C写入内部销售,若B列存在"信达"字符串,则C写入内部采购,否则为空白
时间: 2024-09-24 19:13:55 浏览: 41
Numpy(Pandas)删除全为零的列的方法
在Pandas DataFrame中,你可以通过条件过滤和赋值操作来完成这样的任务。这里是一个简单的示例:
```python
import pandas as pd
# 假设df1是一个DataFrame,已包含'A'和'B'两列
df1 = pd.DataFrame({
'A': ['客户1', '信达', '客户3', '其他'], # 示例A列
'B': ['供应商1', '信达集团', '供应商3', '信达物流'] # 示例B列
})
# 新建C列,并使用np.where函数进行条件判断
df1['C'] = np.where(df1['A'].str.contains('信达'), '内部销售', '')
df1['C'] = df1['C'].fillna(np.where(df1['B'].str.contains('信达'), '内部采购', ''))
# 结果:
# C列将会包含对应的标签:'内部销售', '内部采购' 或者空值 ''
```
这个例子中,`str.contains()`函数用于查找字符串中是否包含特定的子串,`np.where`则根据条件返回不同的值。
阅读全文