Advertisement
cuteSquirrel

data frame of feature

Mar 31st, 2021 (edited)
324
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.80 KB | None | 0 0
  1. import pandas as pd
  2.  
  3. df = pd.DataFrame({
  4.     'A': [101, 102, 99, 100, 105],
  5.     'B': [27, 31, 33, 34, 40],
  6.     'C': [300, 211, 203, 299, 254]
  7. })  
  8.  
  9. def gen_feature(source_df):
  10.  
  11.     feat_x = []
  12.     feat_y = []
  13.  
  14.     for i  , val in source_df.iteritems():
  15.  
  16.         # define feature x and feature y here
  17.         x = val * 10
  18.         y = val + 2
  19.        
  20.         feat_x.append(x)
  21.         feat_y.append(y)
  22.  
  23.     # convert feature into corresponding dataframes
  24.     df_x = pd.DataFrame( zip(*feat_x), columns = [ series.name for series in feat_x])
  25.     df_y = pd.DataFrame( zip(*feat_y), columns = [ series.name for series in feat_y])
  26.  
  27.     return df_x, df_y
  28.  
  29. res1, res2 = gen_feature(df)
  30.  
  31. print("\n", "source df \n", df)
  32. print("\n", "feature x of df \n", res1)
  33. print("\n", "feature y of df \n", res2)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement