Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import pandas as pd
- df = pd.DataFrame({
- 'A': [101, 102, 99, 100, 105],
- 'B': [27, 31, 33, 34, 40],
- 'C': [300, 211, 203, 299, 254]
- })
- def gen_feature(source_df):
- feat_x = []
- feat_y = []
- for i , val in source_df.iteritems():
- # define feature x and feature y here
- x = val * 10
- y = val + 2
- feat_x.append(x)
- feat_y.append(y)
- # convert feature into corresponding dataframes
- df_x = pd.DataFrame( zip(*feat_x), columns = [ series.name for series in feat_x])
- df_y = pd.DataFrame( zip(*feat_y), columns = [ series.name for series in feat_y])
- return df_x, df_y
- res1, res2 = gen_feature(df)
- print("\n", "source df \n", df)
- print("\n", "feature x of df \n", res1)
- print("\n", "feature y of df \n", res2)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement