我有一个 Python 数据框,其中包含字典列表(对于某些行):
In[1]:
cards_df.head()
Out[1]:
card_id labels
0 'cid_1' []
1 'cid_2' []
3 'cid_3' [{'id': 'lid_a', 'name': 'lname_a'}, {'id': 'lid_b', 'name': 'lname_b'}]
4 'cid_4' [{'id': 'lid_c', 'name': 'lname_c'}]
我想创建一个新的数据框,将字典项目列表扩展为单独的行:
card_id label_id label_name
0 cid_3 lid_a lname_a
1 cid_3 lid_b lname_b
2 cid_4 lid_c lname_c
最佳答案
使用pd.Series.str.len
生成适当的值传递给np.repeat
。这又用于重复 df.card_id.values 的值并创建新数据帧的第一列。
然后在 df['labels']
上使用 pd.Series.sum
将所有列表连接成一个列表。这个新列表现在非常适合传递给 pd.DataFrame
构造函数。剩下的就是在每个列名称前面添加一个字符串,并连接到我们上面创建的列。
pd.DataFrame(dict(
card_id=df.card_id.values.repeat(df['labels'].str.len()),
)).join(pd.DataFrame(df['labels'].sum()).add_prefix('label_'))
card_id label_id label_name
0 cid_3 lid_a lname_a
1 cid_3 lid_b lname_b
2 cid_4 lid_c lname_c
<小时/>
设置
df = pd.DataFrame(dict(
card_id=['cid_1', 'cid_2', 'cid_3', 'cid_4'],
labels=[
[],
[],
[
{'id': 'lid_a', 'name': 'lname_a'},
{'id': 'lid_b', 'name': 'lname_b'}
],
[{'id': 'lid_c', 'name': 'lname_c'}],
]
))
关于Python Dataframe包含字典列表,需要使用字典项创建新的dataframe,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45405555/