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')
Pandas使用apply函数给表格添加多列_20
最新推荐文章于 2024-03-10 20:58:53 发布