sorindragan
10/13/2019 - 2:46 PM

Pandas Data Frame manipulation

import pandas as pd
from sklearn.neighbors import KNeighborsRegressor
from sklearn.metrics import mean_squared_error

train = {'author': ['green', 'white'],
         'age': [21, 34],
         'pages': [100, 200],
         }

train = pd.DataFrame(train)
print(train.head())

features = ['pages', 'age']

train_age = train.loc[train['age'].between(25, 34)]

print(train_age)
print(train_age[features])

for i, d in train.iterrows():
    print(d[features])
    age = d['age']
    print(age)