[PYTHON/PANDAS] melt 함수 : id_vars 인자를 사용해 DataFrame 객체의 특정 컬럼 기준으로 데이터 만들기

■ melt 함수의 id_vars 인자를 사용해 DataFrame 객체의 특정 컬럼을 기준으로 데이터를 만드는 방법을 보여준다.

▶ main.py


import pandas as pd

dataFrame1 = pd.DataFrame(
    {
        "first"  : ["John", "Mary"],
        "last"   : ["Doe", "Bo"],
        "height" : [5.5, 6.0],
        "weight" : [130, 150]
    }
)

print(dataFrame1)

"""
  first last  height  weight
0  John  Doe     5.5     130
1  Mary   Bo     6.0     150
"""

dataFrame2 = pd.melt(dataFrame1, id_vars = ["first", "last"])

print(dataFrame2)

"""
  first last variable  value
0  John  Doe   height    5.5
1  Mary   Bo   height    6.0
2  John  Doe   weight  130.0
3  Mary   Bo   weight  150.0
"""

import pandas as pd

dataFrame1 = pd.DataFrame(

{

"first" : ["John", "Mary"],

"last" : ["Doe", "Bo"],

"height" : [5.5, 6.0],

"weight" : [130, 150]

}

)

print(dataFrame1)

"""

first last height weight

0 John Doe 5.5 130

1 Mary Bo 6.0 150

"""

dataFrame2 = pd.melt(dataFrame1, id_vars = ["first", "last"])

print(dataFrame2)

"""

first last variable value

0 John Doe height 5.5

1 Mary Bo height 6.0

2 John Doe weight 130.0

3 Mary Bo weight 150.0

"""

▶ requirements.txt


numpy==2.1.2
pandas==2.2.3
python-dateutil==2.9.0.post0
pytz==2024.2
six==1.16.0
tzdata==2024.2

numpy==2.1.2

pandas==2.2.3

python-dateutil==2.9.0.post0

pytz==2024.2

six==1.16.0

tzdata==2024.2

※ pip install pandas 명령을 실행했다.