如何避免在生成器表達式中添加字符串的錯誤？

Question

我需要遍歷文件夾名稱，然后遍歷圖像，但出現此錯誤。 有人可以告訴我如何避免錯誤嗎？

path = '/.../'

dirs = next(os.walk(path))[1] # get my folder names inside my directory

for i in dirs:
    for img in os.listdir(path+(x for x in dirs)): <------ TypeError: must be str, not generator
        img_path = os.path.join(path,img)  
        print(img_path)

Answer 1

該錯誤來自您試圖向Generator exp添加path的上一行：

path+(x for x in dirs)

您應該使用os.path.join將path連接到目錄名稱：

for dir in dirs:   
   for img in os.listdir(os.path.join(path, dir)):
      ...

Answer 2

import os
path = '/home/'


dirs = next(os.walk(path))[1]  # get folder names inside directory

for i in dirs:
    for img in os.listdir(path+i):
        img_path = os.path.join(path,img)  
        print(img_path)

在下面的行中，您嘗試將生成器對象和字符串path串聯起來。 相反，您可以如上所述使用i本身。

path+(x for x in dirs)

Answer 3

您不需要使用listdir使代碼復雜化。 這個：

import os, os.path
path = '/.../'
for d, _, files in os.walk(path):
    for f in files:
        img_path = os.path.join(d, f)
        print(img_path)

應該足夠了。