在数据框架的列中拆分元组

Question

我有一个数据框df，包含列a中如下所示的元组

|  ID      |  A         |
+----------+------------+
|0         |(1, [a])    |
|1         |(2, [a,b])  |
|2         |(3, [c,a,b])|
+----------+------------+

我想在上面的df中拆分元组，并希望看到我的新数据框架df如下所示。

|  ID      |  A         |    B     |
+----------+------------+----------+
|0         | 1          |  [a]     |
|1         | 2          |  [a,b]   |
|2         | 3          |  [c,a,b] |
+----------+------------+----------+

那么，我如何在上面的数据框df中分割元组呢?

Answer 1

使用str访问器，确保从创建B开始，以避免丢失A中的数据:

df['B'] = df['A'].str[1]
df['A'] = df['A'].str[0]

替代:

df[['A', 'B']] = pd.DataFrame(df['A'].to_list(), columns=['A', 'B'])

输出:

ID  A          B
0   0  1        [a]
1   1  2     [a, b]
2   2  3  [c, a, b]