在PANDAS中每隔n行将数据转置到一列中

时间:2016-09-29 04:22:56

标签: python pandas dataframe reshape transpose

对于研究项目,我需要将网站上的每个人的信息处理成一个excel文件。我已经将我需要的所有东西从网站复制并粘贴到excel文件中的一个列上,然后我使用PANDAS加载了该文件。但是,我需要水平呈现每个人的信息,而不是像现在一样垂直呈现。 例如,这就是我现在所拥有的。我只有一列无组织数据。

df= pd.read_csv("ior work.csv", encoding = "ISO-8859-1")

数据:

0 Andrew
1 School of Music
2 Music: Sound of the wind
3 Dr. Seuss
4 Dr.Sass
5 Michelle
6 School of Theatrics
7 Music: Voice
8 Dr. A
9 Dr. B

我希望每5行转置一次,将数据组织成这种组织格式;下面的标签是列的标签。

Name School Music Mentor1 Mentor2

最有效的方法是什么?

1 个答案:

答案 0 :(得分:4)

如果没有数据缺失,您可以使用numpy.reshape

print (np.reshape(df.values,(2,5)))
[['Andrew' 'School of Music' 'Music: Sound of the wind' 'Dr. Seuss'
  'Dr.Sass']
 ['Michelle' 'School of Theatrics' 'Music: Voice' 'Dr. A' 'Dr. B']]

print (pd.DataFrame(np.reshape(df.values,(2,5)), 
                    columns=['Name','School','Music','Mentor1','Mentor2']))
       Name               School                     Music    Mentor1  Mentor2
0    Andrew      School of Music  Music: Sound of the wind  Dr. Seuss  Dr.Sass
1  Michelle  School of Theatrics              Music: Voice      Dr. A    Dr. B

通过shapepiRSquared生成length array的更一般解决方案除以列数:

print (pd.DataFrame(np.reshape(df.values,(df.shape[0] / 5,5)), 
                    columns=['Name','School','Music','Mentor1','Mentor2']))
       Name               School                     Music    Mentor1  Mentor2
0    Andrew      School of Music  Music: Sound of the wind  Dr. Seuss  Dr.Sass
1  Michelle  School of Theatrics              Music: Voice      Dr. A    Dr. B

感谢{{3}}寻求其他解决方案:

print (pd.DataFrame(df.values.reshape(-1, 5), 
                    columns=['Name','School','Music','Mentor1','Mentor2']))