从数据框中提取特征并将其转换为其他数据框？

Question

我有一个datafarme (my_data:) 如下：

   my_data:        
                    0            ...              16
      TB1     [1, 5, 24, 1]      ...        [0, 0, 0, 31]
      TB2     [7, 4, 13, 1]      ...        [0, 0, 0, 25]
      TB3     [7, 6, 20, 0]      ...        [7, 4, 2, 20]
      ...       .......          ...          ........

可以看出，在每一列中都有一个数字列表，my_data 中总共有 16 列包含数字列表。现在，我想提取这些列表并将它们用作常规列。所以我想要的 my_data 应该是这样的：

   my_data:
             0  1  2   3  ...  60  61 62
      TB1    1, 5, 24, 1  ...  0,  0, 31
      TB2    7, 4, 13, 1  ...  0,  0, 25
      TB3    7, 6, 20, 0  ...  4,  2, 20
      ...       .......   ...   ........

有什么想法吗？

Answer 1

一种简单的方法是沿轴sum 1 并构造一个新的数据框：

pd.DataFrame(df.sum(1).values.tolist(), index=df.index)

     0  1   2  3  4  5  6   7
TB1  1  5  24  1  0  0  0  31
TB2  7  4  13  1  0  0  0  25
...

Answer 2

使用reshape将DataFrame转换为二维数组并通过构造函数创建新的DataFrame：

df = pd.DataFrame(np.array(df.values.tolist()).reshape(len(df), -1), index=df.index)
print (df)
     0  1   2  3  4  5  6   7
TB1  1  5  24  1  0  0  0  31
TB2  7  4  13  1  0  0  0  25
TB3  7  6  20  0  7  4  2  20