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

Question

前提

初歩的な質問になってしまい申し訳ございません。
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の部分についても複数行作成されてしまい、失敗となりました。

Answer 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などのリスト以外は要素が１個の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

Comments

[k4151]

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

[8524ba23]

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

[k4151]

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