pandas のリストを参照したい

Question

以下のようなデータフレームがあります。

id	list
1	[a, b, c]
2	[b, c]
3	[d]

これを以下のようなデータフレームに変換したいです。

id	list1	list2	list3
1	a	b	c
2	b	c	NaN
3	d	NaN	NaN

以下のようなコードを書きましたが、out of range にしまいます。

df['list1'] = df['list'].map(lambda x: x[0])
df['list2'] = df['list'].map(lambda x: x[1])
df['list3'] = df['list'].map(lambda x: x[2])

また、for でも試しましたが時間がかかりすぎてしまいます。

このデータ変換方法についてご教授いただけると幸いです。
よろしくお願いいたします。

Answer 1

listで記述された列を展開するには

df['list'].apply(pd.Series)

でいけます。

import pandas as pd

df = pd.DataFrame({'id':[1,2,3],
                   'list':[['a','b','c'],['d','e'],['f']]})

df = df[['id']].join(df['list'].apply(pd.Series))
#   id  0    1    2
#0   1  a    b    c
#1   2  d    e  NaN
#2   3  f  NaN  NaN

Column名は適当に修正してください

df = df.rename(columns={i: f'list{i+1}' for i in range(3)})
#   id list1 list2 list3
#0   1     a     b     c
#1   2     d     e   NaN
#2   3     f   NaN   NaN

Comments

[退会済みユーザー]

できました。

ありがとう御座います。