from google.colab import drive

drive.mount('/content/drive')

import pandas as pd

from sklearn.preprocessing import LabelEncoder, OrdinalEncoder

import xgboost as xgb

train = pd.read_csv("/content/drive/MyDrive/open (1)/train.csv")
test = pd.read_csv("/content/drive/MyDrive/open (1)/test.csv")
train_wt_columns = train.columns[(train == 'WT').all()].tolist()
test_wt_columns = test.columns[(test == 'WT').all()].tolist()

print('데이터셋에서 값이 모두 WT인 열 개수')
print('trainSet: ', len(train_wt_columns))
print('testSet: ', len(test_wt_columns))
test_df = test.drop(columns=both_wt_columns, errors='ignore')
train_df = train.drop(columns=both_wt_columns, errors='ignore')