데이터의 분리(Q/A)

지도 학습(Supervised Learning) 에서 질문과 그에 적합한 답을 기계에 학습시킬 때, 데이터로부터 질문(x)과 답(y)을 구분해 학습시켜야 한다. 

zip 함수로 분리

q_a = [['질문1', 1], ['질문2', 2], ['질문3', 3]]
x, y = zip(*q_a)
 
# x = ('질문1', '질문2', '질문3')
# y = (1, 2, 3)

DataFrame 으로 분리

import pandas as pd
 
data = [['질문1', 1], ['질문2', 2], ['질문3', 3]]
columns = ['질문컬럼', '답변컬럼']
 
df = pd.DataFrame(data, columns=columns)
x = df['질문컬럼']
y = df['답변컬럼']
print(df)
print(x)
print(y)
 
"""
  질문컬럼  답변컬럼
0  질문1     1
1  질문2     2
2  질문3     3

0    질문1
1    질문2
2    질문3
Name: 질문컬럼, dtype: object

0    1
1    2
2    3
Name: 답변컬럼, dtype: int64
"""

Numpy 로 분리

import numpy as np
 
arr = np.arange(0,16).reshape(4,4)
"""
[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]
 [12 13 14 15]]
"""
 
res = arr[:, :2] 
"""
[[ 0  1]
 [ 4  5]
 [ 8  9]
 [12 13]]
"""
 
res = arr[:, 3]
# [ 3  7 11 15]