python lambda 常见用法

1. 普通直接运算
2. if else
3. if elif else
4. dataframe处理列

# 普通运算
f = lambda x,y: x + y
print(f(3,2))
f = lambda x,y: (x+y, x-y)
print(f(3,2))
print(type(f(3,2)))
# 保留两位小数
f = lambda x : '%.2f'%x
print(f(0.22222))
# ... 等等

# if else
f = lambda x: 'best' if x>80 else 'bad'
print(f(90))

# if elif else
f = lambda x: 'best' if x>80 else('middle' if x>60 else 'bad')
print(f(70))

# dataframe中使用
import pandas as pd
data = pd.DataFrame({'data1' : ['a', 'b', 'c', 'b', 'a'],
    'data2' : ['one', 'one', 'three', 'two', 'one'],
    'num1' : [1,2,3,4,5],
    'num2' : [6,7,8,9,10]})
f = lambda x : 'num2>8' if x>8 else('num2=8' if x==8 else 'num2<8')
data['num2_grade'] = data['num2'].apply(f)
print(data)