python基础-获取指定类型文件、创建excel并获取数据

# 获取指定类型文件路径：
import os

file_path = 'D:\code'
file_path_list = []

for root, dirs, files in os.walk(file_path, topdown=True):
    for file in files:
        if file.endswith('.py'):
            file_path_list.append(os.path.join(root, file))

print(file_path_list)


import openpyxl
from openpyxl import Workbook
import pandas as pd

# 创建excel
workbook = Workbook()
sheet = workbook.active
sheet.title = 'sheet1'
sheet.append(['第一列', '第二列'])
workbook.save('test.xlsx')

# 给excel中添加数据
list_data = ['a', 'b']
def add_data():
    bg = openpyxl.load_workbook('test.xlsx')
    sheet = bg['sheet1']
    for i in range(1, len(list_data)+1):
        sheet.cell(i + 1, 1, list_data[i - 1])
    bg.save('test.xlsx')
add_data()

# 获取excel数据
df = pd.read_excel('test.xlsx', engine='openpyxl', sheet_name='sheet1')
list1 = []
col_data = df.iloc[ :,0]
for col in col_data:
    list1.append(col)
print(list1)