apply()是一种可自定义的函数,可以对Series或DataFrame的行列进行操作并返回结果,可以用于复杂逻辑的实现,针对Series和DataFrame的应用有区别:Series作用每一个元素,不用设行列;DataFrame需要设置行列方向,并作用于其中一种。
(1)Series
df['score'].apply(lambda x: x-3 if x>90 else x)
(2)DataFrame
- # 作用于列元素:aixs=0
- def col(x):
- if x.name='score':
- return x+5
- else:
- return x
-
- df.apply(col, axis=0)
-
-
- # 作用于行元素:axis=1
- def row(x):
- if x['subject']=='lakers':
- a = 1
- else:
- a = 1.2
- return x['score']*a
-
- df.apply(row, axis=1)
(3)传入参数
- # args额外参数
- def score_bias(x, bias):
- if x>90:
- return x+bias
- else:
- return x
-
- df['score'] = df['score'].apply(score_bias, args=(bias,))
(4)传入关键字
- def subject_map(x, **kwargs):
- return kwargs[x]
-
- df['subject_no'] = df['subject'].apply(subject_map, english=0, math=1)
只能作用在DataFrame上,操作对象是每个元素,即接收一个标量元素返回一个标量元素,点对点操作。
- def el_cook(x):
- if isinstance(x, str):
- return 's_'+x
- else:
- return str(x)
-
-
- df.apply(el_cook)
只能作用在Series上。
(1)字典映射
- GENDER_ENCODING = {
- "male": 0,
- "femal": 1
- }
- df['gender_map'] = df['gender'].map(GENDER_ENCODING)
(2)函数映射
map相对于apply无法传参,但是效率高很多。
- # 普通函数
- df['score'],map(np.sqrt)
- df['student'].map(list)
-
- # 自定义函数
- df['score'].map(lambda x:x-3 if x>90 else x)
对series和DataFrame都使用,DataFrame可选择处理的轴方向,默认是列方向。
不支持有降维功能的函数,比如聚合函数min、mean、std。
返回结果与自身形状相同,不改变原数据形状。
- ### 单个函数
- df.transform(np.exp).transform(lambda x: round(x,2))
-
- ### 多个函数
- # 列表形式
- df.transform([np.square, np.sqrt]).transform(lambda x: round(x,2)) # 平方、开平方根(产生多级索引,一级是列名,二级是函数名)
- # 字典形式
- df.transform({'C_0':np.square, 'C_2':[np.square, np.sqrt]}).transform(lambda x:round(x,2)) # 对指定列进行差异化的函数转换
不同于applymap元素级,apply/transform行列级应用,pipe是一个表级应用函数,也称管道函数。
- ### 单个函数
- df.pipe(np.exp).pipe(lambda x:round(x,2))
-
- ### 链式调用
- pi = df.pipe(np.square).\
- pipe(np.multiply, 1.5).\
- pipe(np.add, 8)
-
- ### 特殊传参方式
- def spcl(num, df):
- return df.add(num)
-
- df.pipe((spcl, 'df'), 2) # spcl指定函数,2指定参数
eval()可以通过字符串表达式的方式对series和DataFrame进行计算和解析等操作。
eval()函数有两大优势:
对数据较大的DataFrame对象操作更高效;
对复杂的算术和布尔运算更快速,因为后端计算引擎默认是numexpr
如果数据量较小则没必要用eval,一般当数据量较大超过10000行的时候才建议使用eval()函数进行加速。
eval()支持以下算术操作:
算术运算:除左移(<<)和右移(>>)运算符外的算术运算;
比较操作:包括链式比较,例如,2
布尔运算:例如,df
列表和元组:如[1,2],(1,2);
属性访问:如df.a;
下标表达式:如df[0];
变量评估:如pd.eval("df");
数学函数:如sin,cos,exp等。
eval()不允许使用Python语法:
表达式
数学函数以外的函数调用
is/is not操作
if表达式
lambda表达式
list/set/dict
literal的dict和set表达式
yield表达
生成器表达式
仅包含标量值的布尔表达式
声明
for, while, if
eval()在pandas中有两种函数形式。
- ##### 函数1 #####
- pandas.eval()
- ----------
- 返回:eval解析出来的格式,ndarray,scalar,pandas对象,或None
参数:
expr:指定要被解析的字符串,不能包括任何python的声明 。
parser:指定解析方式,可以是pandas或python,默认为pandas。
engine:后端支持的计算引擎:
None:尝试使用numexpr引擎如果失效则切换到python引擎
numexpr:默认引擎(需要额外安装),可以大幅提高有复杂表达式数据的速度
python:可以像在python中使用eval执行操作一样
target:当expr表达式里有变量赋值时,需要指定变量所在的DataFrame对象。
inplace:如果指定了target,是否对target生效,True代表生效,False则返回target的复制。
- #### 函数2 #####
- dataframe.eval()
- ----------
- 返回:eval解析出来的格式,ndarray,scalar,pandas对象,或None
dataframe.eval()是pandas.eval()的高级封装,可以专门对dataframe对象操作,无需指定target。
参数:
expr和inplace可单独设置,同上。其他参数可通过**kwargs关键字进行设置。
(1)单列变量
- pd.eval("C_4 = (df.C_0>1) & (df.C_2 == 4)", target=df)
- df.eval("C_4 = (df.C_0>1) & (df.C_2 == 4)")
| 0 | 1 | 4 | 2 | 2 | FALSE |
| 1 | 4 | 1 | 1 | 4 | FALSE |
| 2 | 1 | 4 | 4 | 4 | FALSE |
| 3 | 2 | 1 | 4 | 1 | TRUE |
| 4 | 3 | 2 | 1 | 2 | FALSE |
(2)多列变量
- df.eval(
- """
- C_4 = C_0 + C_1
- C_5 = C_1 + C_2
- C_6 = C_2 + C_3
- """
- )
| 0 | 1 | 4 | 2 | 2 | 5 | 6 | 4 |
| 1 | 4 | 1 | 1 | 4 | 5 | 2 | 5 |
| 2 | 1 | 4 | 4 | 4 | 5 | 8 | 8 |
| 3 | 2 | 1 | 4 | 1 | 3 | 5 | 5 |
| 4 | 3 | 2 | 1 | 2 | 5 | 3 | 3 |
(3)局部变量
字符串表达式中可加入局部变量参与计算,通过@前缀标识完成。该前缀方法只能应用于dataframe.eval()函数,对于panda.eval()不生效。
- a = 5
- b = 2
- df.eval("C_4 = C_0 * @a + @b")
| 0 | 1 | 4 | 2 | 2 | 7 |
| 1 | 4 | 1 | 1 | 4 | 22 |
| 2 | 1 | 4 | 4 | 4 | 7 |
| 3 | 2 | 1 | 4 | 1 | 12 |
| 4 | 3 | 2 | 1 | 2 | 17 |
(4)类型解析
注意这里的eval()是python内置的方法,不是pandas提供的函数,无需调用。
- # 字符类型
- a = '[1,2,3]'
- >>> type(a)
- str
-
- # eval释放
- b = eval(a) # 去掉'',并自动转换成里面数据该有的类型
- >>> type(b)
- list