Python：查找名称中仅包含数字的顶级文件夹

Question

我想查找名称中仅包含数字的顶级文件夹。 如果我们有这样的文件夹结构

.
├── Folder1
|   ├── some.file
├── 111
|   ├── some.folder
|   ├── some.file
|   ├── some.file
|   ├── some.file-2
├── 555
|   ├── some.folder
|   ├── some.file

预期结果：找到文件夹“ 111”和“ 555”

这是我的代码：

import os

main_path = 'C:\\Users'
top_folders_list = next(os.walk(main_path))[1]
condition = '111'
if condition in top_folders_list:
   ...do_something...

代码有效，但（仅适用于）文件夹“ 111”。 我应该使用哪种条件来匹配“ 111”，“ 555”以及名称中只有数字的所有其他顶级文件夹？

Answer 1

在字符串对象上使用isnumeric()

for fold in fold_lst:
    if fold.isnumeric():
        print(fold)  

Answer 2

您可以使用正则表达式来过滤条件：

“ \\ d”仅代表数字。

import re

folders = ("123451","8971231")
for folder in folders:
  x = re.findall("\\d", folder)
  result = "".join(x)
  print(result)

那应该给您想要的效果。