时间戳

首页 > 解决方案 > 重塑/透视具有重复键值的键值对数据文件

python - 重塑/透视具有重复键值的键值对数据文件

问题描述

我发现了许多类似标题的问题，但找不到我正在寻找的确切问题。

我有一个这样的数据文件：

title1:A1
title2:A2
title3:A3
title4:A4
title5:A5

title1:B1
title2:B2
title3:B3
title4:B4
title5:B5

title1:C1
title2:C2
title3:C3
title4:C4
title5:C5

title1:D1
title2:D2
title3:D3
title4:D4
title5:D5

使用熊猫我想得到一张这样的桌子：

    title1  title2  title3  title4  title5
0   A1  A2  A3  A4  A5
1   B1  B2  B3  B4  B5
2   C1  C2  C3  C4  C5
3   D1  D2  D3  D4  D5

我的尝试：

import pandas as pd
import numpy as np

df = pd.read_csv('colon_sep.txt',header=None,sep=':')
df.columns = ['title','id']

# for loop method
df2 = pd.DataFrame()
for t in df.title.unique():
    df2[t] = df[df.title == t]['id'].values
df2
# HOW TO GET THIS BY Advanced methods?

我能够使用 for 循环获得所需的表。

有没有更好的方法使用 groupby 或任何其他高级方法？

标签： pythonpandasdataframepivot

解决方案

您可以稍微简化一下代码，以pivot在最后包含一个调用以提高效率：

df = pd.read_csv('colon_sep.txt', sep=':', header=None)
df.insert(2, 2, df.groupby(0).cumcount())  
df = df.pivot(index=2, columns=0, values=1)


print(df)

0 title1 title2 title3 title4 title5
2                                   
0     A1     A2     A3     A4     A5
1     B1     B2     B3     B4     B5
2     C1     C2     C3     C4     C5
3     D1     D2     D3     D4     D5

推荐阅读