将行值添加为数据框中的新列

Question

我有一个如下的数据帧：

A     B     C
a1    b1    c1
a1    b2    c2
a2    b1    c3
a2    b2    c4

我想将其转换为如下内容：

A     B1   B2     
a1    c1   c2     
a1    c1   c2     
a2    c3   c4     
a2    c3   c4     

B中重复的值是列名称，其值在C列中。我希望它们是数据帧的列

事实上，数据集是通过展平一棵树创建的，其中有更多的列，每个列都是一个内部节点。第一列是根，C是叶

以下是我的一些尝试：

cmd = "b1,b2"
cond = ""
for c in cmd.split(","):
    cond += "|" + f" (df['B'] == '{c}') "
    cols = [c] + cols
    new_col = main_df[main_df['B'] == c]['C']
    df[c] = new_col
df = df[eval(cond)]

然而，它不起作用。

Answer 1

尝试：

df.pivot（'A'，columns='B'）

您将只获得两行，但您的四行无论如何都是重复的两行。

    A   C
B       b1  b2
0   a1  c1  c2
1   a2  c3  c4