python(vscode)/폴더명 리스트 가져오기/원하는 경로 지정

특정 경로에 있는 폴더면 리스트를 가져와보자.

 

1. 폴더 경로 정하기

1) 아래 이미지내의 경로의 파일,폴더 이름들을 가져와보자. 

 

2. 코드작성

import os
 
filepath = r"C:\Users\Documents\python_test\stock_study"   
file_list = os.listdir(filepath)
 
print(file_list)

 

3. 결과

  1) 아래 이미지와 같이 리스트에 3가지 네임이 들어가 있다. 

  2) 이미지 파일은 빼고 폴더만 추려보자.

  

 

4. 추가확인 : 폴더만 추리기

  1) 리스트를 가져온 후에 폴더인지 확인하면된다.

  2) 그 방법은 경로로 만들어 경로가 유효한지 판단하면 된다.   

 

4-1. 코드작성

import os
 
filepath = r"C:\Users\fusgh\Documents\python_test\stock_study"   
file_list = os.listdir(filepath)
 
forder_list =[] 
 
for item in file_list:
    sub_folder = os.path.join(filepath, item)  # path 합치기
    if os.path.isdir(sub_folder): # 폴더 여부 확인
        forder_list.append(item)
 
print(forder_list)

4-2 결과

  1) 폴더명인 1과 2만 출력되었다.