Pandas实现excel的IF函数功能
1、创建数据框
dic = {'地市': '廊坊','廊坊','廊坊','张家口','张家口','张家口','组号':1,2,1,2,1,2,'数量':11,12,13,14,15,16}
p_city = pd.DataFrame(dic)
print(p_city)
输出：
p_city'编号' = np.where(p_city'组号' == 1, '男', '女') #
print(p_city)
输出：
2.新建列
列名与原列明 不相同 是新建
p_city'性别' = np.where(p_city'组号' == 1, '男', '女') #
print(p_city)
输出：
3.指定位置新建列
思路：先新建，再同列明
insert(loc, column, value, allow_duplicates=False)
在哪里：
loc： 插入列的索引。第一列是 0。
column： 赋予新列的名称。
value： 新列的值数组。
allow_duplicates： 是否允许新列名匹配现有列名。默认值为假。
dic = {'地市': '廊坊','廊坊','廊坊','张家口','张家口','张家口','组号':1,2,1,2,1,2,'数量':11,12,13,14,15,16}
p_city = pd.DataFrame(dic)
p_city.insert(1, '性别2', "") # 位置，列明，数据
p_city'性别2' = np.where(p_city'组号' == 1, '男', '女') #
print(p_city)
输出：
4.进阶（if嵌套）
dic = {'地市': '廊坊','廊坊','廊坊','张家口','张家口','张家口','组号':1,2,1,2,1,2,'数量':11,12,13,14,15,16}
p_city = pd.DataFrame(dic)
p_city'是否去办公室' = np.where(p_city'组号' == 1, np.where(p_city'数量' > 13, '是', '否'), '否')
print(p_city)
输出：
