# -*- coding: utf-8 -*-
"""
Created on Wed Apr  7 17:14:08 2021

@author: Sim
"""
'''
와이드 포맷과 롱포맷으로 변환
wideform1.txt는 환자번호, 성별, 최초내원시 측정값, 1주 및 2주 후의 측정값이 
정리된 파일임. 이 파일을 long format으로 변환하고 다시 wide format으로 변환하는
문제
'''

import pandas as pd

wide1 = pd.read_csv("D:\HTEX\Pythonbk\codesdata\wideform1.txt", sep="\s+", header=0)
long1 = pd.melt(wide1, id_vars=['patient','gender'], value_vars=['week0', 'week1', 'week2'],
                var_name='week', value_name='response')
print('long format by melt: \n', long1)

wide2 = long1.pivot(index=['patient','gender'], columns='week', values='response')
print('wide format by pivot: \n', wide2)

wide3 = wide2.reset_index()
print('wide format by pivot: \n', wide3)

print('original wide format:\n', wide1)

####### wide_to_long

long5 = pd.wide_to_long(wide1, stubnames='week', 
                        i=['patient','gender'], j='order')
long5.reset_index()
print(long5)
wide6 = pd.read_csv("D:\HTEX\Pythonbk\codesdata\wideform2.txt", sep="\s+", header=0)
long6 = pd.wide_to_long(wide6, stubnames='week', 
                        i=['patient','gender'], j='order')