And 연산 Data pit
import pandas as pd
import numpy as np
import matplotlib.pyplot as pit
# 방법1 이중리스트 [[0,0,0],[0,1,0]....] 행으로 들어감
# 방법2 딕셔너리 {컬럼명: [0,0,0,......]} 열로 들어감
data_dic = {'A':[0,0,1,1,1,0,1,0], 'B':[0,1,0,1,0,0,1,1], 'A and B':[0,0,0,1,0,0,1,0]}
df = pd.DataFrame(data_dic)
문제 , 답 분리 (A ,B 분리)
X = df.loc[:,'A':'B']
#문제 X는 대문자
y = df.iloc[:,-1]
#답 y는 소문자
데이터 확인
X.shape
#(8, 2)
y.shape
#(8,)훈련용 X_train, 테스트용 X_test 분리 및 확인
X_train = X.iloc[:6]
X_train.shape
#(6, 2)X_test = X.iloc[6:,:]
X_test.shape
#(2, 2)훈련용 y_train, 테스트용 y_test 분리 및 확인
y_train = y.iloc[6:]
y_train.shape
#(6,)y_test = y.iloc[6:]
y_test.shape
#(2,)
hello!