In python/pandas, 1) How to add new variable (the lead_value
as image) ? 2) How to filter the dataframe which type
include ":" ?
import pandas as pd
ori_df=pd.DataFrame()
ori_df=pd.DataFrame([['a','1'],['w:','z'],['t','6'],['f:','z'],['a','2']],
columns=['type','value']
)
CodePudding user response:
import pandas as pd
ori_df = pd.DataFrame(
[['a','1'],['w:','z'],['t','6'],['f:','z'],['a','2']],
columns=['type','value']
)
ori_df['lead_value'] = ori_df['value'].shift(-1).fillna(0)
ori_df_filtered = ori_df[ori_df['type'].apply(lambda t: ':' in t)]
print(ori_df_filtered)
prints
index | type | value | lead_value |
---|---|---|---|
1 | w: | z | 6 |
3 | f: | z | 2 |