Python使用pandas每三行转列

Question

我有一个文本文件，其中的数据每 3 行重复一次。可以说它是哈希、目录、子目录。数据如下所示:

a3s2d1f32a1sdf321asdf
Dir_321321
Dir2_asdf
s21a3s21d3f21as32d1f
Dir_65465
Dir2_werq
asd21231asdfa3s21d
Dir_76541
Dir2_wbzxc
....

我创建了一个 python 脚本，它获取数据并且每 3 行创建列:

import pandas as pd

df1 = pd.read_csv('RogTest/RogTest.txt', delimiter = "\t", header=None)
df2 = df1[df1.index % 3 == 0]
df2 = df2.reset_index(drop=True) 
df3 = df1[df1.index % 3 == 1]
df3 = df3.reset_index(drop=True)
df4 = df1[df1.index % 3 == 2]
df4 = df4.reset_index(drop=True)
df5 = pd.concat([df2, df3], axis=1)
df6 = pd.concat([df5, df4], axis=1)

#Rename columns
df6.columns = ['Hash', 'Dir_1', 'Dir_2']
#Write to csv
df6.to_csv('RogTest/RogTest.csv', index=False, header=True)   

这工作正常，但我很好奇是否有更有效的方法来做到这一点，也就是更少的代码？

Answer 1

您可以使用:

df_final = pd.DataFrame(np.reshape(df.values,(3, df.shape[0]/3)))
df_final.columns = ['Hash', 'Dir_1', 'Dir_2']

输出:

                    Hash       Dir_1       Dir_2
0  a3s2d1f32a1sdf321asdf  Dir_321321   Dir2_asdf
1   s21a3s21d3f21as32d1f   Dir_65465   Dir2_werq
2     asd21231asdfa3s21d   Dir_76541  Dir2_wbzxc