【Python】——Excel

import pandas as pd


# 1. 表格行列互换
def exchange_row_axi(root):
    # data_ = pd.read_csv(r'./train.csv', encoding='utf-8')     # 读取csv格式
	df = pd.read_excel(root)        # 读取xls格式
    df.head()
    print('df： ', df)
    df.T.apply(pd.Series.to_list, axis=0)
    print('df： ', df)
    df.to_csv('exchange_row_axi.csv')
    return df


exchange_row_axi('./example.xls')


# 2. 提取某几列到另一个新csv文件
def extract_row(root):
    df = pd.read_excel(root)
    print(df.head(10))

    df = df.loc[:, ['a']]
    df.to_csv('extract_row.csv')
    return df


extract_row('./example.xls')


# 3. 将Excel某一列的数据存放到一个列表里
def excel_one_line_to_list(root):
    # df = pd.read_excel(root)
    # print(df.head(10))
    df = pd.read_csv(r'./train.csv', encoding='utf-8')     # 读取csv格式

    df_li = df.values.tolist()
    result = []
    for s_li in df_li:
        result.append(s_li[0])
    # print(result)
    print(type(result))
    return result


excel_one_line_to_list('./example.xls')


# 4. 将Excel好几列的数据存放到一个列表里
def excel_several_line_to_list(root):
    result = []
    for i in range(3):
        print(i)
        df = pd.read_excel(root, usecols=[i],
                           names=None)
        df_li = df.values.tolist()
        for s_li in df_li:
            result.append(s_li[0])
        print(result)
    return result


excel_several_line_to_list('./example.xls')


# 4. 将列表保存到本地Excel，一个数字一个格子
def list_to_excel():
    a = ['1', '2', '3', '4']
    df = pd.DataFrame(a, columns=['a'])     # 列名
    df.to_excel("list_to_excel.xlsx", index=False)
    return df


list_to_excel()