想通过python对制定文件夹内所有文本文件的内容进行按文件名和日期区别的归集

Question

想通过python对制定文件夹内所有文本文件的内容进行按文件名和日期区别的归集，日期一列，内容股票名称一列，代码一列，来自的文件名一列，来自文件日期生成一列，（所有文件的内容都是固定的“股票名称（股票代码）”格式），归集写入Excel。然后每天接着之前的内容写入。

Answer 1

可以使用 Python 的 os 模块来遍历文件夹中的文本文件，然后使用正则表达式提取文件中的股票名称和股票代码。

为了将数据写入 Excel，可以使用 pandas 库的 DataFrame 类和 ExcelWriter 类。首先，你需要创建一个空的 DataFrame，然后逐个遍历文件并将每个文件中提取的信息添加到 DataFrame 中。最后，使用 ExcelWriter 将 DataFrame 写入 Excel。

import os
import re
import pandas as pd

# 遍历文件夹中的文件
folder_path = "path/to/folder"
file_list = []
for file_name in os.listdir(folder_path):
    if file_name.endswith(".txt"):
        file_list.append(file_name)

# 创建 DataFrame
columns = ['date','name','code','file_name','file_date']
df = pd.DataFrame(columns=columns)

# 遍历文件并提取信息
for file_name in file_list:
    file_path = os.path.join(folder_path, file_name)
    with open(file_path, "r") as file:
        file_content = file.read()
        match = re.search(r"(.*)\((.*)\)", file_content)
        if match:
            name, code = match.groups()
            date = pd.datetime.now()
            file_date = os.path.getmtime(file_path)
            df = df.append({'date':date,'name':name,'code':code,'file_name':file_name,'file_date':file_date},ignore_index=True)

# 将 DataFrame 写入 Excel
with pd.ExcelWriter("output.xlsx") as writer:
    df.to_excel(writer, sheet_name='Sheet1')