import pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv)

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
import sklearn.linear_model as sk
full_data_set = pd.read_csv('../input/nflplaybyplay2015.csv', low_memory=False)

Pass_Plays = full_data_set.loc[full_data_set.PlayType == 'Pass']
Sack_Plays = full_data_set.loc[full_data_set.PlayType == 'Sack']
P_S_data = pd.concat([Pass_Plays, Sack_Plays])

good_columns = ['Drive', 'qtr', 'down', 'TimeUnder', 'TimeSecs', 'PlayTimeDiff', 'yrdline100', 'ydstogo']
good_columns += ['ScoreDiff', 'PosTeamScore', 'DefTeamScore']
good_columns += ['Sack']
uncleaned_data = P_S_data[good_columns]

uncleaned_data.qtr.unique()