Python DataFrame：枢轴行作为列

Question

我有来自不同电台的原始文件。当我将它们组合成一个数据框时，我看到三列具有匹配的 id 和名称以及不同的组件。我想将其转换为数据框，其中名称条目成为列名称 代码：

df = 
      id                      name      component
0      1             Serial Number          103
1      2              Station Name           DC
2      1             Serial Number          114
3      2              Station Name           CA
4      1             Serial Number          147
5      2              Station Name           FL

预期答案：

new_df = 
      Station Name      Serial Number     
0     DC                  103
1     CA                  114
2     FL                  147

我的答案：

# Solution1
df.pivot_table('id','name','component')
name 
NaN NaN NaN NaN
# Solution2
df.pivot(index=None,columns='name')['component']
name 
NaN NaN NaN NaN

我没有得到想要的答案。有什么帮助吗？

Answer 1

首先，您必须使用相同ID的每2行进行每2行，之后您可以使用枢轴表。

import pandas as pd

df = pd.DataFrame({'id': ["1", "2", "1", "2", "1", "2"],
                    'name': ["Serial Number", "Station Name", "Serial Number", "Station Name", "Serial Number", "Station Name"],
                   'component': ["103", "DC", "114", "CA", "147", "FL"]})
new_column = [x//2+1 for x in range(len(df))]
df["id"] = new_column

df = df.pivot(index='id',columns='name')['component']

Answer 2

如果您的序列号就在站名称之前：

df_ = df.pivot(columns='name', values='component').groupby(df.index // 2).first()

print(df_)

name Serial Number Station Name
0              103           DC
1              114           CA
2              147           FL