IT TIP

Pandas DataFrame을 목록 목록으로 변환

itqueen 2020. 11. 26. 20:33
반응형

Pandas DataFrame을 목록 목록으로 변환


다음과 같은 팬더 데이터 프레임이 있습니다.

admit   gpa  gre  rank   
0  3.61  380     3  
1  3.67  660     3  
1  3.19  640     4  
0  2.93  520     4

이제 다음과 같은 팬더의 행 목록을 얻고 싶습니다.

[[0,3.61,380,3], [1,3.67,660,3], [1,3.19,640,4], [0,2.93,520,4]]   

어떻게하니?


A는 전화, 가장 빠른 방법이 될 것입니다 방법이 내장되어 tolist.valuesNP 배열 :

df.values.tolist()

[[0.0, 3.61, 380.0, 3.0],
 [1.0, 3.67, 660.0, 3.0],
 [1.0, 3.19, 640.0, 4.0],
 [0.0, 2.93, 520.0, 4.0]]

다음과 같이 할 수 있습니다.

map(list, df.values)

편집 : 버전 0.23.0부터 사용되지 않습니다 .as_matrix

데이터 프레임 에서 기본 제공 values또는 to_numpy(권장 옵션) 메서드를 사용할 수 있습니다 .

In [8]:
df.to_numpy()

Out[8]:
array([[  0.9,   7. ,   5.2, ...,  13.3,  13.5,   8.9],
   [  0.9,   7. ,   5.2, ...,  13.3,  13.5,   8.9],
   [  0.8,   6.1,   5.4, ...,  15.9,  14.4,   8.6],
   ..., 
   [  0.2,   1.3,   2.3, ...,  16.1,  16.1,  10.8],
   [  0.2,   1.3,   2.4, ...,  16.5,  15.9,  11.4],
   [  0.2,   1.3,   2.4, ...,  16.5,  15.9,  11.4]])

numpy 배열이 아닌 목록을 명시 적으로 원하면 .tolist()다음을 추가하십시오 .

df.to_numpy().tolist()

참고 URL : https://stackoverflow.com/questions/19585280/convert-pandas-dataframe-into-list-of-lists

반응형