[Python]파이썬 판다스(pandas) DataFrame를 리스트로 변환하는 방법

pandas DataFrame를 리스트로 변환하는 방법에 대해 알아봅니다. 리스트로 변환하려면 tolist()메소드를 사용합니다.

딕셔너리 자료형으로 객체를 하나 생성 후 DataFrame으로 만들었습니다.

import pandas as pd

products = {'Item': ['아이폰', '갤럭시노트20', '안드로이드'],
            'Price': [800000, 1200000, 500000]
            }
print(type(products))

df = pd.DataFrame(products, columns=['Item', 'Price'])

DataFrame의 값을 리스트로 Convert 하는 방법

values.tolist()메소드를 사용하여 값만 리스트로 변환합니다.

df = pd.DataFrame(products, columns=['Item', 'Price'])

val_list = df.values.tolist()
print(type(val_list))
print(val_list)


#실행결과
<class 'dict'>
[['아이폰', 800000], ['갤럭시노트20', 1200000], ['안드로이드', 500000]]
<class 'list'>

DataFrame의 칼럼을 리스트로 변환하는 방법

columns.values.tolist()메서드를 사용하여 변환합니다.

df = pd.DataFrame(products, columns=['Item', 'Price'])
column_list = df.columns.values.tolist()
print(type(column_list))
print(column_list)

#실행결과
<class 'list'>
['Item', 'Price']

컬럼과 값을 함께 리스트로 변환하는 방법

df = pd.DataFrame(products, columns=['Item', 'Price'])

item_list = df.columns.values.tolist() + df.values.tolist()
print(type(item_list))
print(item_list)


#실행결과
<class 'list'>
['Item', 'Price', ['아이폰', 800000], ['갤럭시노트20', 1200000], ['안드로이드', 500000]]

특정 값만 리스트로 변환하는 방법

df = pd.DataFrame(products, columns=['Item', 'Price'])

item_list = df['Item'].values.tolist()
#item_list.append('플레이스테이션')
print(item_list)


#실행결과
['아이폰', '갤럭시노트20', '안드로이드']

[전체 샘플 코드]

import pandas as pd

products = {'Item': ['아이폰', '갤럭시노트20', '안드로이드'],
            'Price': [800000, 1200000, 500000]
            }
print(type(products))

df = pd.DataFrame(products, columns=['Item', 'Price'])

products_list = df.values.tolist()
print(type(products_list))
print(products_list)



column_list = df.columns.values.tolist()
print(type(column_list))
print(column_list)



item_list = df.columns.values.tolist() + df.values.tolist()
print(type(item_list))
print(item_list)


item_list = df['Item'].values.tolist()
#item_list.append('플레이스테이션')
print(item_list)

[다른글 더보기]

playground.naragara.com/671