使用pipe可以像水流一样,有顺序的执行dataframe
,而且代码会更清晰
示例代码
import pandas as pd
import numpy as np
def change_1(value, v_1):
value['change_1'] = v_1 # 新建一列"change_1",并赋值为v_1
return value
def change_2(value, v_2):
value['change_2'] = v_2 # 新建一列"change_2",并赋值为v_2
return value
if __name__ == '__main__':
df = pd.DataFrame(np.random.random(size=(10, 2)))
pipe_return = df.pipe(change_1, "a").pipe(change_2, "b")
print(pipe_return)