質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

ただいまの
回答率

88.07%

pandasのDataFrameでlist型の要素を複数行に展開する方法

解決済

回答 1

投稿

  • 評価
  • クリップ 1
  • VIEW 1,603

score 1

前提

初歩的な質問になってしまい申し訳ございません。
pythonでDataFrame内のlist型の要素を複数行に展開して、新たにDataFrameを作成したいと考えております。
下記の様なDataFrameから、infoの要素を展開させたいです。
その際に、list内のdictをカラムとして、新たに列を追加する方法をご教授いただけると幸いです。

id menu_id info
1 10 [{"name":"bread","price":100}, {"name":"rice","price":200}]
2 20 Nan
3 30 [{"name":"curry","price":300}]

やりたいこと

listを要素を複数行に展開し、その中にあるdictをカラムとして新たな列を追加したいです。
下記の様な結果をイメージしております。

id menu_id name price
1 10 "bread" 100
1 10 "rice" 200
2 20 Nan Nan
3 30 "curry" 300

試したこと

tmp_df = df.set_index('id')['info'].apply(pd.Series).unstack().reset_index(1).rename(columns={0:'info'})
res = df[['id', 'menu_id']].merge(tmp_df, on='id')


listを複数行に展開することは出来たのですが、infoがNanの部分についても複数行作成されてしまい、失敗となりました。

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

ここにより詳細な情報を記載してください。

  • 気になる質問をクリップする

    クリップした質問は、後からいつでもマイページで確認できます。

    またクリップした質問に回答があった際、通知やメールを受け取ることができます。

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 過去に投稿した質問と同じ内容の質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

回答 1

checkベストアンサー

+1

以下のような感じでできます。
参考:Pandas column of lists, create a row for each list element

import pandas as pd
import numpy as np

df = pd.DataFrame({'id':[1,2,3], 'menu_id':[10,20,30], 'info':[[{"name":"bread","price":100}, {"name":"rice","price":200}], np.nan, [{"name":"curry","price":300}]]})
print(df)
#   id  menu_id                                               info
#0   1       10  [{'name': 'bread', 'price': 100}, {'name': 'ri...
#1   2       20                                                NaN
#2   3       30                  [{'name': 'curry', 'price': 300}]

# 'info'列の要素をSeriesに
def to_sr(info, key):
    # リストはSeriesに
    if type(info) is list:
        lst = [e[key] for e in info]
        return pd.Series(lst)

    # NaNなどのリスト以外は要素が1個のSeriesに
    return pd.Series([info])

# Seriesに展開
series = {}
for key in ['name', 'price']:
    f = lambda row: to_sr(row['info'], key)
    series[key] = df.apply(f, axis=1).stack().reset_index(level=1, drop=True)

# 各SeriesをDataFrameに結合し元dfに結合
dfs = pd.DataFrame(series)
df = df.drop('info', axis=1).join(dfs).reset_index(drop=True)
print(df)
#   id  menu_id   name  price
#0   1       10  bread  100.0
#1   1       10   rice  200.0
#2   2       20    NaN    NaN
#3   3       30  curry  300.0

投稿

編集

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2020/07/29 16:36

    ご回答頂きありがとうございます。
    やはり関数を作成しそちらで処理させるのがよさそうですね。
    仮にinfo内の要素が多数ある場合に、別で関数を作成せずに実装できる方法をご存じでしたら、ご教授いただけると幸いです。

    キャンセル

  • 2020/07/29 16:47

    関数をlambdaで作りこめばできます。回答修正しました。

    キャンセル

  • 2020/07/29 17:55

    迅速に回答頂きありがとうございます。
    実装することができました。

    キャンセル

15分調べてもわからないことは、teratailで質問しよう!

  • ただいまの回答率 88.07%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる

関連した質問

同じタグがついた質問を見る