注意是双重目录,
import os
def GetFileFromDirectory(path: str) -> list[str]:
files = os.listdir(path)
for i in range(len(files)):
files[i] = path + "\\" + files[i]
x = os.listdir(files[i])[0]
files[i] = files[i] + "\\" + x
return files
files = GetFileFromDirectory(directorypath)
增加cityid的版本,
def GetFileFromDirectory(path: str) -> list[str]:
cityids = []
files = os.listdir(path)
for i in range(len(files)):
cityid = files[i].split("_")[-2]
cityids.append(cityid)
# print(f'cityid = {cityid}')
files[i] = path + "\\" + files[i]
x = os.listdir(files[i])[0]
files[i] = files[i] + "\\" + x
return [files, cityids]