将具有混合整数的列拆分为多列的元组

Question

我有一个DF   a     b   0  (0,1) 11  1     (1,2)2  2     3所需的输出是：    w  x  y  z0   0  1  1  01   1  0  1  22   2  0  3  3问题是，元组可以有多个不同的长度。以下 tolist() 仅适用于长度为 2 的元组，不适用于混合列。df[['w', 'x']]=pd.DataFrame(df['a'].tolist(), index=df.index) 有任何想法吗？提前致谢。

Answer 1

想法是添加元组 if 标量然后创建新列：

def f(col):

    return pd.DataFrame([x if isinstance(x, tuple) else (x, ) 

                        for x in col]).fillna(0).astype(int)

df[['w', 'x']]=df.pop('a').pipe(f)

df[['y', 'z']]=df.pop('b').pipe(f)

print (df)

   w  x  y  z

0  0  1  1  0

1  1  0  1  2

2  2  0  3  0

更通用的解决方案concat：

dfs= [pd.DataFrame([x if isinstance(x, tuple) else (x, ) for x in df.pop(c)], 

                    index=df.index) for c in df.columns]

df = pd.concat(dfs, axis=1, ignore_index=True).fillna(0).astype(int)

print (df)

   0  1  2  3

0  0  1  1  0

1  1  0  1  2

2  2  0  3  0

Answer 2

您可以转换为，然后str转换strip为()split,

>>> df[['w', 'x']] = pd.DataFrame(df.pop('a')

                                    .astype(str)

                                    .str.strip('(/)')

                                    .str.split(',')

                                    .tolist()).fillna(0).astype(int)

>>> df[['y', 'z']] = pd.DataFrame(df.pop('b')

                                    .astype(str)

                                    .str.strip('(/)')

                                    .str.split(',')

                                    .tolist()).fillna(0).astype(int)

>>> df

   w  x  y  z

0  0  1  1  0

1  1  0  1  2

2  2  0  3  0