python - 解压数据帧值中的集合并复制到行中

Question

有一个 pandas 数据框，其中有一列可能是任意长度的集合:

n = np.nan
stack1 = pd.DataFrame.from_dict(  
         {'letter1': ['a','b','c','y'],
        'letter2': [ 'o','p', 'q', 'y'],
      'overlap': [ {'v'},{'c'}, {'c'}, {'v', 'c'}]
       })
stack1.reset_index(inplace=True, drop=True)

从这个数据框中，我如何解压集合的内容并从每个解压的元素创建新行？如果该解决方案也适用于列表和元组等其他容器，那就太好了。

期望的结果:

Answer 1

试试这个:

In [32]: col_to_unpack = 'overlap'

In [33]: df = stack1.copy()

In [34]: pd.DataFrame({
    ...:      col:np.repeat(df[col].values, df[col_to_unpack].str.len())
    ...:      for col in df.columns.difference([col_to_unpack])
    ...: }).assign(**{col_to_unpack:np.concatenate(df[col_to_unpack].map(list).values)})[df.columns.tolist()]
    ...:
Out[34]:
  letter1 letter2 overlap
0       a       o       v
1       b       p       c
2       c       q       c
3       y       y       c
4       y       y       v