栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 面试经验 > 面试问答

根据if-elif-else条件创建新列

面试问答 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

根据if-elif-else条件创建新列

要使上面列出的一些方法正式化:

创建一个对数据框的行进行操作的函数,如下所示:

def f(row):    if row['A'] == row['B']:        val = 0    elif row['A'] > row['B']:        val = 1    else:        val = -1    return val

然后通过

axis=1
选项将其应用于您的数据框:

In [1]: df['C'] = df.apply(f, axis=1)In [2]: dfOut[2]:   A  B  Ca  2  2  0b  3  1  1c  1  3 -1

当然,这不是矢量化的,因此在缩放到大量记录时,性能可能会不佳。不过,我认为它更具可读性。特别是来自SAS背景。

编辑

这是向量化版本

df['C'] = np.where(    df['A'] == df['B'], 0, np.where(    df['A'] >  df['B'], 1, -1))


转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/637179.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号