👩💻 Develope/Python
[Pandas] 데이터프레임 리스트로 변환하기
heywantodo
2023. 10. 4. 11:11
728x90
반응형
[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)
728x90
반응형