pandas中的单列多列运算

作者: 麻辣阁 分类: python 发布时间: 2018-12-10 14:05

单列运算

在Pandas中,DataFrame的一列就是一个Series, 可以通过map来对一列进行操作:

df['col2'] = df['col1'].map(lambda x: x**2)

其中lambda函数中的x代表当前元素。可以使用另外的函数来代替lambda函数,例如:

define square(x):
    return (x ** 2)

df['col2'] = df['col1'].map(square)

多列运算

要对DataFrame的多个列同时进行运算,可以使用apply,例如col3 = col1 + 2 * col2:

df['col3'] = df.apply(lambda x: x['col1'] + 2 * x['col2'], axis=1)

其中x带表当前行,可以通过下标进行索引。

参考:


发表评论

电子邮件地址不会被公开。 必填项已用*标注