python - 如何将值中包含列表的数据框转换为每个级别都作为 python 中的单个列值的大数据框？

Question

我的数据框如下所示:

mydata = [{'col_A' : 'A', 'col_B': [1,2,3]},
      {'col_A' : 'B', 'col_B': [7,8]}]
pd.DataFrame(mydata)


col_A   col_B
    A   [1, 2, 3]
    B   [7, 8]

如何拆分列表中的值并创建如下所示的数据框:

col_A   col_B
A   1
A   2
A   3
B   7
B   8

Answer 1

试试这个:

pd.DataFrame([{'col_A':row['col_A'], 'col_B':val} 
               for ind, row in df.iterrows()
               for val in row['col_B']])

您也许还可以使用 apply() 函数做一些巧妙的事情，但在我的头脑中，我可以想到如何做。