본문 바로가기

BLOG/데이터분석

[데이터분석] 데이터프레임 : 특정 칼럼(열)을 리스트로 바꾸는 방법

 

데이터프레임에서 특정 칼럼(열)에 있는 데이터를 리스트의 형식으로 만들고 싶을 수 있는데요,

이러한 경우 간단하게 for문을 이용해서 리스트로 만들 수 있답니다.

 

 

먼저, 아래와 같은 데이터프레임(df1)이 있다고 가정해보겠습니다.

 

# <df1>
	item	price	count
0	water	1000	1
1	milk	2000	5
2	coffee	3000	2
3	ade	4000	1
4	juice	5000	3
5	tea	6000	2

 

이 데이터프레임에서 만약 'item' 칼럼에 있는 데이터들을 리스트로 만들고 싶다면 아래와 같이 for문을 사용해서 코드를 입력하면 됩니다.

 

item_list = []
for i in df1['item'] :
    item_list.append(i)

 

먼저 비어있는 리스트를 만들고, for문에서 df의 'item' 칼럼에서 데이터를 하나씩 순서대로 리스트에 추가되도록 하였습니다. append() 함수는 리스트에 데이터를 추가해주는 함수로 사용되었습니다.

 

그 결과, item_list 리스트를 출력해보면 다음과 같이 'item' 칼럼에 있던 데이터들이 잘 저장되었음을 확인할 수 있습니다.

 

['water', 'milk', 'coffee', 'ade', 'juice', 'tea']