如何将多标签数据集拆分为多行，每行有一个标签？

Question

我有一个熊猫数据框，看起来像这样

Feature      Class
text1        [label1, label2]
text2        [label2, label3]

最好的方法是什么

Feature      Class
text1        label1
text1        label2
text2        label2
text2        label3

注意：在初始表中，类值是包含标签的Python列表

Answer 1

您可以将^{}与^{}一起使用，并通过^{}将列表展平：

from itertools import chain

df = pd.DataFrame({
    'Feature' : df['Feature'].values.repeat(df['Class'].str.len()),
    'Class' : list(chain.from_iterable(df['Class'].values.tolist()))
})
print (df)
  Feature   Class
0   text1  label1
1   text1  label2
2   text2  label2
3   text2  label3