如何在 python 中将行值转换为列名和与列值相同的列名？

Question

我想将行值转换为列名，这些列名将是列值。 我已经尝试过使用 pivot，但它没有提供所需的 output。

Data:
col1   col2   col3 col4
x1      y1      z1   a1
x1      y1      z1    a2
x1      y1      z1    a3

我试过如下：

Data.pivot(columns='Col4', values='col4')

Output:
a1  a2   a3
a1 NAN   NAN
NAN a2   NAN
NAN  NAN  a3

Desired output:

col1 col2 col3  a1  a2 a3
x1     y1   z1  a1  a2 a3

Answer 1

您可以使用pivot ，但您需要指定index参数以保留“col1”、“col2”、“col3”。 一旦你这样做了，你可以稍微清理一下 dataframe 以获得你想要的结果。

out = (df.pivot(index=["col1", "col2", "col3"], columns="col4", values="col4")
         .rename_axis(columns=None)
         .reset_index())

print(out)
  col1 col2 col3  a1  a2  a3
0   x1   y1   z1  a1  a2  a3

脚步

• pivot(...) : pivot dataframe 根据需要。 这使得“col1”、“col2”和“col3”的MultiIndex 。 然后是“col4”值的实际列和值
• rename_axis(columns=None) : pivot使列成为具有名称的Index object。 我发现这个名字让人们无法调整 dataframe 的表示方式。 在这里，我从列Index中删除了名称，以便可以表示 OP 的期望
• 如前所述， reset_index() - pivot生成“col1”、“col2”和“col3”的MultiIndex 。 我们在这里使用reset_index()来获取这些值并将其插入到DataFrame的实际数据中，而不是作为MultiIndex