Python ler o nome de arquivos dentro de uma pasta

Estou aprendendo python e queria sabe como faz para o python entrar em uma certa pasta(path) e no caso ele printar todos os nomes de arquivos da pasta.

Exemplo:

  • joao.txt
  • maria.xslx
  • augusto.rar
import os
os.listdir('/Caminho/Para/O/Diretorio')
1 curtida

Obrigado, caso eu queira que ele entre em 2 pastas e procurar por cada arquivo com a extensão txt como seria?
Tenho um diretório raiz: reservatorio
Dentro dele tem: pasta1 e pasta2.

# main.py
import os

def encontraArquivosEmPastaRecursivamente(pasta='.', extensao):
    arquivosTxt = []
    caminhoAbsoluto = os.path.abspath(pasta)
    for pastaAtual, subPastas, arquivos  in os.walk(caminhoAbsoluto):
        arquivosTxt.extend([os.path.join(pastaAtual,arquivo) for arquivo in arquivos if arquivo.endswith('.txt')])
    return arquivosTxt

print encontraArquivosEmPastaRecursivamente('.', '.txt')

Minhas pastas:

.
├── main.py
└── reservatorio
    ├── pasta1
    │   ├── p1.foo
    │   └── p1.txt
    └── pasta2
        ├── p2.bar
        └── p2.txt

Resultado:

['/Users/lucas/Desktop/reservatorio/pasta1/p1.txt',
 '/Users/lucas/Desktop/reservatorio/pasta2/p2.txt']
1 curtida