Pythonにてディレクトリ名,ファイル名の配列を取得しcsvを作成したい

Question

pythonにてosライブラリを使用し、ディレクトリ名とファイル名をまとめる作業をしていますがうまくできません。

下記の状態から、
メインディレクトリ
├20201105
│├filename1
│└filename2
├20201104
│└filename3
└20201103
└filename4

下記に示すようなcsvを生成するスクリプトを作成したいです。
20201105　filename1
20201105　filename2
20201104　filename3
20201103　filename4

import os
import csv
import datetime

col1 = list()
col2 = list()
for current,subfolders,subfiles in os.walk("メインディレクトリ"):
    print([current])
    print([subfolders])
    print([subfiles])
    print("-------------------------------------------------------")
#     if len([subfolders])==[]:


        col1.append(current[-3:])
        col2.append(subfiles)
print(col1)
print(col2)

試したこと

ここに問題に対して試したことを記載してください。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

メインディレクトリの直下にあるディレクトリ（dir）を取得し、
for文でdirの直下のファイルを取得する（ディレクトリもヒットしますが...）という流れで作成してみましたがどうでしょうか？
csvファイルに保存する仕様にしています。

import os
import csv
import datetime

col1 = list()
col2 = list()

main_dir = "メインディレクトリ"
# print(os.listdir(main_dir)) # 直下のファイル、ディレクトリ取得
files = os.listdir(main_dir)
dir = [f for f in files if os.path.isdir(os.path.join(main_dir, f))] # ディレクトリのみ取得
# print(dir)

with open('./sample.csv', 'w') as f:
  writer = csv.writer(f) # sample.csvファイルに書き込み

  for current in dir:
    for subfiles in os.listdir(main_dir + current):
      print(current,subfiles)
      writer.writerow([current,subfiles]) # sample.csvファイルに書き込み
      col1.append(current)
      col2.append(subfiles)

print("-------------------------------------------------------")
print(col1)
print(col2)

Comments

[KOPOKER]

ありがとうございます！
この仕様で問題ないです！