jack06215

[pandas] extract rows using np.r_()

Jun 22nd, 2020
116
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.42 KB | None | 0 0
  1. import pandas as pd
  2. import dask.dataframe as dd
  3. import numpy as np
  4. import os
  5. from sklearn.datasets import make_classification
  6.  
  7. df = make_classification(n_samples = 2000, n_features = 10)
  8. df_raw = pd.DataFrame(df[0], columns = ['var1', 'var2', 'var3', 'var4', 'var5', 'var6', 'var7', 'var8', 'var9', 'var10'])
  9. df_raw['class'] = df[1]
  10.  
  11. cmd = '2, 3, 5:10, 100:104'
  12. cmd = str('np.r_[') + cmd + str(']')
  13. df_raw.iloc[eval(cmd),:]
Add Comment
Please, Sign In to add comment