Pandas数据帧的列表列表

Question

嗨我正在尝试查找列表,给定一个listID我可以找到拥有它的用户,并且给定一个UserID我可以找到该用户的所有列表.

数据采用以下格式:

[['34', '345'],
['12', '23,534,34'],
['1', '13,42']]

我想要的是一个pandas数据框,如下所示:

UserID, ListID
34, 345
12, 23
12, 534
12, 34
 1, 13
 1, 42

我的想法是将第二个字符串分成"逗号"分隔的列表,但是从那里我被卡住了.有什么建议？

Answer 1

您应该在将数据提供给数据框构造函数之前清理它们.这是一个简单的脚本:

import pandas as pd

data = [['34', '345'],
['12', '23,534,34'],
['1', '13,42']]

new_data = []
for row in data:
    x, yvals = row
    for y in yvals.split(','):
        new_data.append([x,y])

df = pd.DataFrame(new_data, columns=['UserID', 'ListID'])