Pythonを用いて旅行データを１行にまとめたデータを作る

Question

前提・実現したいこと

Pythonを用いて旅行データを１行にまとめたデータを作りたいです．
例
|旅行者|出発地|目的地|
|１|大阪|東京|
|１|東京|北海道|
|１|北海道|大阪|
|２|大阪|福岡|
|２|福岡|大阪|
というデータを

|旅行者|出発地|経由地１|経由地２|到着地|
|１|０|０|０|０|
|２|０|０|０│０|
というデータフレームを作り

|旅行者|出発地|経由地１|経由地２|到着地|
|１|大阪|東京|北海道|大阪|
|２|大阪|福岡|０│大阪|

といった旅行者１人の旅行を１行にまとめたいです．

発生している問題・エラーメッセージ

|旅行者|出発地|経由地１|経由地２|到着地|
|０|大阪|大阪|大阪|大阪|
|０|大阪|大阪|大阪|大阪|
という形になっておりうまくできません

該当のソースコード

コード

import pandas as pd
df=pd.read_excel('旅行データ')
df1=pd.DataFrame({'旅行者':[0,0], '出発地':[0,0],'訪問地１':[0,0],'訪問地２':[0,0],'到着地':[0,0]},index=[0,1],dtype=str)
for i in range(5):
    for j in range(2):
        for k in range(1,5):
            df1.at[df1.index[j],df1.columns[k]]=df.at[df.index[i],'目的地']
            if df.at[df.index[i],'旅行者']==df.at[df.index[i+1],'旅行者']:
                df1.at[df1.index[j],df1.columns[k+1]]=df.at[df.index[i+1],'目的地']
                if df.at[df.index[i],'旅行者']!=df.at[df.index[i+1],'サンプルID']:
                    df1.at[df1.index[j+1],df1.columns[k]]=df.at[df.index[i+1],'旅行者']

補足情報（FW/ツールのバージョンなど）

Anconda Navigatorでjupyter Notebookを使用しています．

Comments

[yamato_user]

dfの出力を記述してください

[tatsumi-1]

dfの出力は一番最初の表です．分かりにくくてすいません．

[tatsumi-1]

dfの出力は一番最初の表の状態です．

[KojiDoi]

３箇所以上を経由することはありえないという前提でしょうか？

[tatsumi-1]

実際には経由します．今回はデータが多いのでfor文を回すときに時間を短くするため経由地を２箇所にしています．

Answer 1

とりあえず、こんな感じでどうでしょう。

import pandas as pd

df = pd.DataFrame([[1,'大阪','東京'],
                   [1,'東京','北海道'],
                   [1,'北海道','大阪'],
                   [2,'大阪','福岡'],
                   [2,'福岡','大阪']],
                  columns=['旅行者','出発地','目的地'])

grouped_df = df.groupby('旅行者')
departure_df = grouped_df['出発地'].apply(lambda d:d.reset_index(drop=True)).unstack()
arrival_df = grouped_df['目的地'].apply(lambda d: d.iat[-1])

result = departure_df.join(arrival_df)
result.columns = ['出発地', '経由地１', '経由地２', '到着地']
print(result)
#    出発地 経由地１  経由地２ 到着地
#旅行者
#1    大阪   東京   北海道  大阪
#2    大阪   福岡  None  大阪

---

【追記】

ループを使う方法

import pandas as pd

df = pd.DataFrame([[1,'大阪','東京'],
                   [1,'東京','北海道'],
                   [1,'北海道','大阪'],
                   [2,'大阪','福岡'],
                   [2,'福岡','大阪']],
                  columns=['旅行者','出発地','目的地'])

max_trip_size = df.groupby('旅行者').size().max()

df1 = pd.DataFrame(columns=range(max_trip_size + 1))
for member,d in df.groupby('旅行者'):
    # 出発地+経由地を追加
    df1 = df1.append(d['出発地'].reset_index(drop=True).rename(member))
    # 到着地を追加
    df1.loc[member, max_trip_size] = d['目的地'].iat[-1]

# カラム名を修正
df1 = df1.add_prefix('経由地')
df1 = df1.rename(columns={'経由地0':'出発地', '経由地{}'.format(max_trip_size):'到着地'})
print(df1)
#  出発地 経由地1 経由地2 到着地
#1  大阪   東京  北海道  大阪
#2  大阪   福岡  NaN  大阪

Comments

[tatsumi-1]

回答ありがとうふございます．
実際には旅行者がかなり多くいるので最初のdfのデータからデータフレームに代入していく方法が知りたいです．

[magichan]

とりあえずループ処理にてデータを追記していく方法を記載しました。

> 実際には旅行者がかなり多くいるので
であればなおさらループ使わないほうがよいと個人的には思うのですが。