Pandas使用apply函数给表格添加多列_20

import numpy as np
import pandas as pd

"""
本节主要介绍如何使用apply函数给表格添加多列
"""

# 导入csv
path = './test.csv'
df = pd.read_csv(path)


# df.apply方法添加一列
# 定义一个函数，返回一列
def get_wendu_type(x):
    if x['bWendu'] > 33:
        return '高温'
    elif x['bWendu'] < -10:
        return '低温'
    else:
        return '常温'


df.loc[:,'wendu_type'] = df.apply(get_wendu_type,axis=1)  # 注意这里要设置axis=1,这里是series的index或者column,而且axis=1就是列
# 查看温度类型的计数
wendu_type = df['wendu_type'].value_counts()

# df.apply方法添加多列
# 定义一个函数，返回多列
def many_columns(x):
    new_a,new_b = df['a']+df['b'],df['a']-df['b']
    return new_a,new_b

df[['new_a','new_b']] = df.apply(many_columns,axis=1,result_type='expand')