用pandas将excel的指定行数据取出并转换为列

用pandas将excel的指定行数据取出并转换为列

转换前：

转换后：

# -*- encoding=UTF-8 -*-
__author__ = 'wjj1982'
__date__ = '2019/7/6 12:35'
__product__ = 'PyCharm'
__filename__ = 'pandas-row_to_column'


import pandas as pd
import os

pd_excel = pd.read_excel('test-bak.xlsx')
new_pd = []
column_new = []
for row in range(0, pd_excel.shape[0]):
    # 取每一行的2-66列数据，注意下面这个取法最后是带列头的字典形式
    new_pd.append(pd_excel.iloc[row][2:66])
    for x in range(0, 64):
        column_new.append(str(pd_excel.iloc[row][0]) + '-' + str(pd_excel.iloc[row][1]))

# concat将new_pd列表连接成新df
pd.concat(new_pd).to_excel('wjj.xlsx')

# 读取后新增一列，因为不会直接写入，哈哈哈
pd_excel2 = pd.read_excel('wjj.xlsx')
pd_excel2['aau-id'] = column_new
pd_excel2.to_excel('wjj2.xlsx')

# 最后删除过渡文件wjj.xlsx
if os.path.exists('wjj.xlsx'):
    os.remove('wjj.xlsx')