问题遇到的现象和发生背景
用代码块功能插入代码,请勿粘贴截图
import pandas as pd
import os
import pyautogui
import time
#查找未下载的文件名
def find_undownloadcode():
# 1.利用OS库遍历文件夹所有的文件名
downloaded_filenames = os.listdir("F:\\gupiaoziliao\\managerstock\\")
# 2.读取excel所有的股票代码
pd1 = pd.DataFrame(pd.read_excel("F:\\gupiaoziliao\\沪深A股.xlsx"))
all_stockcodes = list(pd1['证券代码'])
# 3.删除已下载的文件名
i = 0
while i < len(all_stockcodes):
if all_stockcodes[i] in downloaded_filenames:
del all_stockcodes[i]
i -= 1
i += 1
return all_stockcodes
undownload_codes = find_undownloadcode()
我能想到的解答思路
我能想到的是先将函数运行结果存入到excel,用的时候直接从excel中调用,但是感觉有点麻烦,不知道有没有其他的方法,不用把数据导出,就能达到这种效果的
我想要达到的结果
利用上述代码,得出未下载的代码,后面一直调用undownload_codes列表里面的数据即可,而不用重复运行find_undownloadcode函数