[Pandas] 데이터프레임 리스트로 변환하기

[Pandas] 데이터프레임 리스트로 변환하기

데이터 프레임을 리스트로 변환하여 처리해야 할때,

리스트로 변환하는 방법은 여러가지가 있다.

 

먼저 샘플 df를 생성한다.

import pandas as pd

df = {
    'key': ['A', 'B', 'C', 'D'],
    'value': [1, 2, 3, 4]
}

df = pd.DataFrame(df, columns=['key','value'])

 

칼럼 선택 후 리스트로 변환하기

데이터 프레임의 각 칼럼은 시리즈(Series)라는 데이터 타입으로 이루어져있다.

시리즈는 to_list() 함수를 이용해 리스트로 변환하거나, list() 함수를 이용해서 리스트로 변경할 수 있다.

df['key'].to_list()

['A', 'B', 'C', 'D']

list(df['value'])

[1, 2, 3, 4]

 

행을 리스트로 추출하기

iloc를 이용하면 특정 인덱스를 기준으로 값을 추출할 수 있다.

추출되는 값이 하나라면, 아래와 같이 쉽게 리스트로 변환할 수 있다.

df.iloc[0].to_list()

['A', 1]

 

array로 변경하기

데이터 프레임이 array로 변환된다.

pandas 1.4 이상의 버전에서는 array() 함수를

pandas 1.4 보다 버전이 낮다면 __array__() 함수를 사용해야한다.

df.array()
df.__array__()

array([['A', 1],
       ['B', 2],
       ['C', 3],
       ['D', 4]], dtype=object)

컬럼이 여러개일 때, 특정 컬럼만 지정해서 array로 변환이 가능하다. 

df = {
    'key': ['A', 'B', 'C', 'D'],
    'value': [1, 2, 3, 4],
    'lower' : ['a', 'b', 'c', 'd']
}

df = pd.DataFrame(df, columns=['key','value','lower'])

df[['key','lower']].array()

array([['A', 'a'],
       ['B', 'b'],
       ['C', 'c'],
       ['D', 'd']], dtype=object)

 

참고 : https://aplab.tistory.com/entry/%ED%8C%8C%EC%9D%B4%EC%8D%AC-%EB%8D%B0%EC%9D%B4%ED%84%B0%ED%94%84%EB%A0%88%EC%9E%84-%EB%A6%AC%EC%8A%A4%ED%8A%B8