Obtener todos los archivos en un directorio de Python: Aumenta la eficiencia de tu lista de directorios

Name

Oluwaseun Adeojo

Updated on 17/8/2023

Python, un lenguaje de programación versátil y potente, se utiliza ampliamente para diversas tareas, incluyendo la gestión de archivos y directorios. Una tarea común que los desarrolladores de Python suelen encontrar es la necesidad de listar todos los archivos en un directorio. Este artículo te guiará a través del proceso de utilizar Python para obtener todos los archivos en un directorio, centrándose en la función os.listdir y cómo utilizarla de manera eficiente.

La función os.listdir es parte del módulo os de Python, el cual proporciona una forma portátil de utilizar funcionalidades dependientes del sistema operativo. Permite listar todos los archivos y directorios en el directorio especificado. Sin embargo, no se trata solo de listar archivos; también exploraremos cómo filtrar estos resultados, listar directorios de forma recursiva y trabajar con diferentes tipos de archivos como JSON y CSV.

¿Quieres crear rápidamente visualizaciones de datos a partir de un DataFrame de Python Pandas sin escribir código?

PyGWalker es una biblioteca de Python para el análisis exploratorio de datos con visualización. PyGWalker (opens in a new tab) puede simplificar tu flujo de trabajo de análisis y visualización de datos en Jupyter Notebook, convirtiendo tu DataFrame de pandas (y DataFrame de polars) en una interfaz de usuario de estilo Tableau para la exploración visual.

(opens in a new tab)

1. Utilizando os.listdir para obtener todos los archivos en un directorio

La función os.listdir es una herramienta sencilla pero poderosa en el arsenal de Python. Devuelve una lista que contiene los nombres de las entradas en el directorio especificado por la ruta. Aquí tienes un ejemplo básico de cómo utilizar os.listdir:

import os
 
def listar_archivos(directorio):
    return os.listdir(directorio)
 
print(listar_archivos('/ruta/a/tu/directorio'))

Esto imprimirá una lista de todos los archivos y directorios en el directorio especificado. Sin embargo, es importante tener en cuenta que os.listdir no lista archivos de forma recursiva, es decir, no lista los archivos en los subdirectorios del directorio especificado. A continuación, veremos cómo listar archivos de forma recursiva en el próximo segmento.

2. Listando archivos y directorios de forma recursiva

Para listar archivos y directorios de forma recursiva, es decir, para incluir archivos en todos los subdirectorios, podemos utilizar la función os.walk. La función os.walk genera los nombres de archivos en un árbol de directorios recorriendo el árbol ya sea de arriba hacia abajo o de abajo hacia arriba.

Aquí tienes cómo puedes utilizar os.walk para listar todos los archivos en un directorio y sus subdirectorios:

import os
 
def listar_archivos_recursivamente(directorio):
    for raiz, directorios, archivos in os.walk(directorio):
        for archivo in archivos:
            print(os.path.join(raiz, archivo))
 
listar_archivos_recursivamente('/ruta/a/tu/directorio')

Esto imprimirá las rutas de todos los archivos en el directorio especificado, incluyendo aquellos en sus subdirectorios. La parte os.path.join(raiz, archivo) se utiliza para obtener la ruta completa del archivo.

3. Filtrando archivos por extensión

En algunos casos, es posible que quieras listar solo archivos de un cierto tipo, es decir, archivos con una cierta extensión. Por ejemplo, es posible que solo quieras listar archivos JSON o CSV. Puedes hacer esto agregando una condición simple al código de listado de archivos:

import os
 
def listar_archivos_por_extension(directorio, extension):
    for raiz, directorios, archivos in os.walk(directorio):
        for archivo in archivos:
            if archivo.endswith(extension):
                print(os.path.join(raiz, archivo))
 
listar_archivos_por_extension('/ruta/a/tu/directorio', '.json')

Esto imprimirá las rutas de todos los archivos JSON en el directorio especificado y sus subdirectorios. Puedes modificar fácilmente este código para listar archivos con cualquier otra extensión, como '.csv' para archivos CSV.

En la siguiente parte del artículo, discutiremos cómo listar tipos específicos de archivos, como imágenes.

4. Listando tipos específicos de archivos: Imágenes y vídeos

A menudo, es posible que necesites listar tipos específicos de archivos, como imágenes o vídeos, en un directorio. Python proporciona una forma sencilla de hacer esto verificando la extensión del archivo. Aquí tienes cómo puedes listar todos los archivos de imagen en un directorio:

import os
 
def listar_archivos_de_imagenes(directorio):
    for raiz, directorios, archivos in os.walk(directorio):
        for archivo in archivos:
            if archivo.endswith(('.jpg', '.png', '.jpeg')):
                print(os.path.join(raiz, archivo))
 
listar_archivos_de_imagenes('/ruta/a/tu/directorio')

Esto imprimirá las rutas de todos los archivos de imagen (JPEG y PNG) en el directorio especificado y sus subdirectorios. Puedes modificar fácilmente este código para listar archivos con cualquier otra extensión, como '.mp4' o '.avi' para archivos de vídeo.

De manera similar, para vídeos, puedes utilizar el mismo enfoque, pero con extensiones de archivo de vídeo:

import os
 
def listar_archivos_de_video(directorio):
    for raiz, directorios, archivos in os.walk(directorio):
        for archivo in archivos:
            if archivo.endswith(('.mp4', '.avi', '.mov')):
                print(os.path.join(raiz, archivo))
 
listar_archivos_de_video('/ruta/a/tu/directorio')

Esto listará todos los archivos de vídeo (MP4, AVI y MOV) en el directorio especificado y sus subdirectorios. Este enfoque es bastante flexible y se puede adaptar para listar cualquier tipo de archivo simplemente cambiando las extensiones de archivo en el método endswith.

5. Listado avanzado de directorios: Filtrado y uso de técnicas de inteligencia artificial generativa

Las capacidades incorporadas de Python para listar directorios son bastante poderosas, pero también existen muchas bibliotecas y herramientas de terceros que pueden ayudarte a realizar tareas más avanzadas. Por ejemplo, es posible que desees filtrar los resultados de tu búsqueda según ciertos criterios, o utilizar técnicas de inteligencia artificial generativa para automatizar algunos aspectos del proceso de listado de directorios.

Una de estas herramientas es el módulo glob, que proporciona una función para crear listas de archivos a partir de búsquedas con comodines en un directorio. Esto puede ser especialmente útil cuando deseas listar archivos que coinciden con un patrón determinado. Aquí tienes un ejemplo:

import glob
 
def listar_archivos_segun_patron(directorio, patron):
    return glob.glob(f'{directorio}/{patron}')
 
print(listar_archivos_segun_patron('/ruta/a/tu/directorio', '*.json'))

Esto mostrará una lista de todos los archivos JSON en el directorio especificado. Puedes modificar fácilmente el patrón para que coincida con cualquier otro tipo de archivo.

En cuanto a la inteligencia artificial generativa, es un campo en constante evolución que tiene muchas aplicaciones potenciales en el listado de directorios. Por ejemplo, podrías utilizar un modelo de inteligencia artificial generativa para predecir los tipos de archivos en un directorio en función de su nombre u otros metadatos, o para generar etiquetas descriptivas para los archivos según su contenido. Sin embargo, estos son temas avanzados que van más allá del alcance de este artículo.

Conclusión y Lecturas Adicionales

En este artículo, hemos cubierto muchos aspectos sobre cómo utilizar Python para obtener todos los archivos de un directorio. Hemos visto la función básica os.listdir, cómo listar archivos de forma recursiva con os.walk, cómo filtrar archivos por extensión y cómo listar tipos específicos de archivos como imágenes y videos. También hemos tocado algunos temas avanzados como el uso del módulo glob para la coincidencia de patrones y las aplicaciones potenciales de la inteligencia artificial generativa en el listado de directorios.

Sin embargo, ¡siempre hay más por aprender! Si estás interesado en profundizar en estos temas, te recomiendo consultar la documentación oficial de Python, así como recursos como Stack Overflow y Towards Data Science. Y, por supuesto, la mejor manera de aprender es practicando, así que no dudes en empezar a experimentar con estas técnicas en tus propios proyectos.

Preguntas frecuentes

¿Cómo puedo listar todos los archivos de un directorio utilizando Python?

Puedes utilizar la función os.listdir para listar todos los archivos de un directorio. Esta función devuelve una lista que contiene los nombres de las entradas en el directorio especificado por la ruta.

¿Cuál es la sintaxis para usar os.listdir?

La sintaxis para usar os.listdir es bastante simple. Solo necesitas importar el módulo os y luego llamar a os.listdir(ruta), donde ruta es la ruta al directorio que deseas listar.

¿Puedo filtrar los resultados de os.listdir para mostrar solo archivos con ciertas extensiones?

Sí, puedes filtrar los resultados de os.listdir para mostrar solo archivos con ciertas extensiones. Puedes hacer esto agregando una condición que verifique si el nombre del archivo termina con la extensión deseada.