AI办公自动化:根据Excel某列内容批量新建文件夹

2024-09-18 14:08:40 浏览数 (1)

工作任务:根据excel表格中J列“证券名称”来批量新建文件夹

在deepseek中输入提示词:

你是一个Python编程专家,写一个Python脚本,具体步骤如下:

读取Excel文件"F:AI自媒体内容AI炒股做过AI算法备案的上市公司.xlsx"J列“证券名称”,根据这些证券名称在文件夹“F:研报下载”中批量新建文件夹。

注意:每一步都要输出信息到屏幕上

Deepseek的回复和代码:

当然!下面是一个Python脚本,它将读取Excel文件中的“证券名称”列,并在指定文件夹中为每个证券名称创建一个新的文件夹。每一步都会输出信息到屏幕上。

import os

import pandas as pd

# 定义文件路径

excel_file_path = r"F:AI自媒体内容AI炒股做过AI算法备案的上市公司.xlsx"

output_folder_path = r"F:研报下载"

# 读取Excel文件

print(f"开始读取Excel文件: {excel_file_path}")

try:

df = pd.read_excel(excel_file_path, usecols=['证券名称'])

print("Excel文件读取成功!")

except Exception as e:

print(f"读取Excel文件时出错: {e}")

exit(1)

# 获取证券名称列表

security_names = df['证券名称'].tolist()

print(f"共读取到 {len(security_names)} 个证券名称。")

# 创建文件夹

print(f"开始在 {output_folder_path} 中创建文件夹...")

for security_name in security_names:

folder_path = os.path.join(output_folder_path, security_name)

try:

os.makedirs(folder_path, exist_ok=True)

print(f"成功创建文件夹: {folder_path}")

except Exception as e:

print(f"创建文件夹 {folder_path} 时出错: {e}")

print("所有文件夹创建完成!")

代码说明:

导入必要的库:

os 用于文件和目录操作。

pandas 用于读取Excel文件。

定义文件路径:

excel_file_path 是Excel文件的路径。

output_folder_path 是目标文件夹的路径,用于存放新建的文件夹。

读取Excel文件:

使用 pandas.read_excel 读取Excel文件中的“证券名称”列。

如果读取失败,程序会输出错误信息并退出。

获取证券名称列表:

将“证券名称”列转换为列表。

创建文件夹:

遍历证券名称列表,为每个证券名称在目标文件夹中创建一个新的文件夹。

使用 os.makedirs 创建文件夹,exist_ok=True 表示如果文件夹已经存在则不会报错。

如果创建文件夹失败,程序会输出错误信息。

输出信息:

每一步都会输出信息到屏幕上,方便跟踪脚本的执行情况。

运行脚本:

确保你已经安装了 pandas 库,如果没有安装,可以使用 pip install pandas 进行安装。

将脚本保存为 .py 文件并运行。

这样,你就可以根据Excel文件中的证券名称在指定文件夹中批量创建文件夹了。

程序运行结果:

0 人点赞