Чи є спосіб, яким я можу використовувати glob в каталозі, щоб отримати файли з певним розширенням, але лише ім'я самого файлу, а не весь шлях?
Чи є спосіб, яким я можу використовувати glob в каталозі, щоб отримати файли з певним розширенням, але лише ім'я самого файлу, а не весь шлях?
Відповіді:
Це може допомогти комусь:
names = [os.path.basename(x) for x in glob.glob('/your_path')]
map(os.path.basename, glob.glob("your/path"))
Повертає ітерабель із усіма іменами та розширеннями файлів.
os.path.basename працює для мене.
Ось приклад коду:
import sys,glob
import os
expectedDir = sys.argv[1] ## User input for directory where files to search
for fileName_relative in glob.glob(expectedDir+"**/*.txt",recursive=True): ## first get full file name with directores using for loop
print("Full file name with directories: ", fileName_relative)
fileName_absolute = os.path.basename(fileName_relative) ## Now get the file name with os.path.basename
print("Only file name: ", fileName_absolute)
Вихід:
Full file name with directories: C:\Users\erinksh\PycharmProjects\EMM_Test2\venv\Lib\site-packages\wheel-0.33.6.dist-info\top_level.txt
Only file name: top_level.txt
Я продовжую переписувати рішення для відносного глобінгу (зокрема, коли мені потрібно додати елементи до zip-файлу) - саме так воно зазвичай виглядає.
# Function
def rel_glob(pattern, rel):
"""glob.glob but with relative path
"""
for v in glob.glob(os.path.join(rel, pattern)):
yield v[len(rel):].lstrip("/")
# Use
# For example, when you have files like: 'dir1/dir2/*.py'
for p in rel_glob("dir2/*.py", "dir1"):
# do work
pass
Якщо ви шукаєте файл CSV:
file = [os.path.basename(x) for x in glob.glob(r'C:\Users\rajat.prakash\Downloads//' + '*.csv')]
Якщо ви шукаєте файл EXCEL:
file = [os.path.basename(x) for x in glob.glob(r'C:\Users\rajat.prakash\Downloads//' + '*.xlsx')]