(mapReduce와 비슷한) dataframe.apply를 사용하여 팬다에서 각 행을 여러 행으로 펼치십시오.

Question

다음은 간단한 문제입니다. 여행의 시작 및 끝 위치가있는 DataFrame이 있습니다. 각 스테이션에 대한 DataFrame을 끝내기를 원합니다. 도착 횟수와 출발 횟수입니다.

은 내가 한 행 출력 여러 행, 에 걸릴 후 감소 단계에서 모든 행을 통해 집계 할 수있는 지도 상에 맵리 듀스와 같은 워크 플로우에 익숙입니다.

내가 지금 가지고있는 코드는 다음과 같습니다. 작동하지 않습니다. 내 마지막 DataFrame으로 원하는 것은

import pandas as pd 
import numpy as np 

def expand_row(row): 
    return pd.Series(
    { 'station': [row['start_station'], row['end_station']], 
     'departures': [1, 0], 
     'arrivals': [0, 1], 
    }, 
) 

trips = pd.DataFrame({ 
    'start_station': ['a', 'c'], 
    'end_station': ['b', 'a'], 
}) 

expanded = df.apply(expand_row, axis=1) 
aggregated = expanded.groupby('station').aggregate(np.sum) 

는

desired_df = pd.DataFrame({ 
    'station': ['a', 'b', 'c'], 
    'departures': [1, 0, 1], 
    'arrivals': [1, 1, 0] 
}) 
desired_df.index = desired_df.pop('station') 

많은 감사합니다.

Answer 1

import pandas as pd 
trips = pd.DataFrame({ 
    'start_station': ['a', 'c'], 
    'end_station': ['b', 'a'], 
}) 
trips.apply(pd.value_counts).fillna(0) 

결과는 다음과 같습니다

end_station start_station 
a   1    1 
b   1    0 
c   0    1