工作任务:根据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文件中的证券名称在指定文件夹中批量创建文件夹了。
程序运行结果: