Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from sklearn.preprocessing import StandardScaler
- def transform_data(data):
- scaler = StandardScaler()
- set_1 = ['UniqueCarrier', 'FlightNum','Origin', 'Dest', 'Diverted']
- for typi in set_1:
- data[data[typi].isnull()] = 'nan'
- for typi in set(data.columns) - set(set_1):
- data[data[typi].isnull()] = 0
- pd.DataFrame(data=scaler.fit_transform(data[typi]))
- pd.get_dummies(data)
- return data
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement