pandas生成excel

pandas生成excel及解析

import pandas as pd

# 生成一列excel
def generate_excel_one():
    company_name_list = ['腾讯', '阿里巴巴', '百度']
    df = pd.DataFrame(company_name_list, columns=['company_name'])
    df.to_excel("company_name.xlsx", index=False)

# 生成两列excel
def generate_excel_two():
    company_name_list = [['腾讯', '北京'], ['阿里巴巴', '杭州'], ['百度', '北京']]
    df = pd.DataFrame(company_name_list, columns=['company_name', 'location'])
    df.to_excel("company_name.xlsx", index=False)

# 解析excle到list
def read_excel_to_list():
    df = pd.read_excel("company_name.xlsx", index_col=None, header=None)  
    # df = pd.read_excel("company_name.xlsx", engine="openpyxl", index_col=None, header=None)  # 低版本用这个

    df_list = df.values.tolist()
    for line in df_list:
        print(line)

if __name__ == '__main__':
    generate_excel_two()
    read_excel_to_list()