ファイルを（ファイル名と前半部分が一致する）フォルダへの移動

Question

pythons初心者です。
pythonでファイル名を（ファイル名と前半部分が一致する）フォルダへの移動をするコードを考えております。

【状況】
・未分類のフォルダ（下記のコードだと’invoice2'フォルダ）に分類前のPDF ファイルが格納されております
・PDFファイル名は「1001_aaa.pdf」，「2132_abcd.pdf」のように前半部分は「****_」の形式（数字4文字にアンダーバー）にしております
・分類先のフォルダは、フォルダ名が「1001_㈱××商事」、「2131_㈲×××」のようになっております
・（ファイル名の前半部分の数字4文字）と（フォルダ名の前半部分の数字4文字）が一致するものがあります

【実現したいこと】
・ファイル名の前半の数字4文字である「****」箇所と前半部分が一致するフォルダに格納したいです
・例だとファイル：「1001_aaa.pdf」をフォルダ名「1001_㈱××商事」に格納したいです

【実行したこと】
下記のようなコードを作成しました

import shutil
import os

#file名の取得
files=os.listdir('./ocr/invoice2')

#fileを１つずつ取り出す
for file in files:
    file_name=os.path.splitext(os.path.basename(file))
    file_name=file_name[0] 
    print(file_name)
    path1=f'./ocr/invoice2/{file_name}.pdf'
    path2=f'./ocr/invoice3/{file_name[:4]}/{file_name}.pdf' #ここがうまくいかない
    os.rename(path1,path2) 

【うまくいかない理由（推定）】
・たとえばフォルダ名が「1001_㈱××商事」のフォルダを指定する際に
f'./ocr/invoice3/{file_name[:4]}/'
のようにしてますが、これだと　フォルダ名が「1001」のようにしないとならないのかと思います。
そのため、 　f'./ocr/invoice3/{file_name[:4]}+{*}/' のようにワイルドカードを利用してフォルダ名を指定しようと思いましたがうまくいきませんでした。

ネットなどで調べているのですが、うまい解決方法が見当たらないです。
アドバイスをもらえますと助かります

よろしくお願いいたします

Answer 1

Pythonで条件を満たすパスの一覧を再帰的に取得するglobの使い方
glob --- Unix 形式のパス名のパターン展開

# ディレクトリとファイル作成
!mkdir 1101_商事
!mkdir 2131_有限会社
!mkdir ocr
%cd ocr
!mkdir invoice
%cd /content
!touch ocr/invoice/1101_1.pdf
!touch ocr/invoice/2131_1.pdf

# サンプルコード
import glob
import os

files = glob.glob('./ocr/invoice/*.pdf') # ファイル一覧

dirs = os.listdir()　#　ディレクトリ一覧

for f in files:
    filename = os.path.basename(f)
    num = filename.split('_')[0]　# ファイル名の先頭部分を抽出
    for d in dirs:
        if num in d: # ディレクトリの名前の先頭ととファイルの数字が一致した場合
            os.rename(f,d+'/'+filename) #　ファイル移動

Colaboratoryで実施しました。

こんな感じで、移動してみてはどうでしょうか？

Comments

[kuma07]

大変ありがとうございます。これがやりたかったことです。
このようなシンプルなコードが書きたいです。
大変勉強になりました。
また、よろしくお願いいたします。

Answer 2

他の方の回答通りですが、

1001_abcde.pdf
2012_xyz.pdf

などとというファイルをそれぞれ

1001_あいうおえ
2012_かきくけこ

というフォルダに入れたいのですよね？

であれば、人間が手動でやる時にどうやって入れる先を決めるかですよね。
通常であれば、

ファイル名の先頭4桁を覚えて、フォルダ名の一覧を見て先頭4桁が合致するところに入れる。

とすると思いますので、これをそのままコードにすればいいかと思います。

Comments

[kuma07]

ありがとうございます。
考え方勉強になります。
またよろしくお願いいたします。

Answer 3

1. 前半部の文字列、というのをとってきておいて、
2. フォルダ名のリストを取ってきて、
3. そのリストから、その文字列で始まるものを探す、

というのをまず考えればどうでしょう

Comments

[kuma07]

アドバイスありがとうございます。
アドバイスを踏まえて作成しましたら完成しました。
下記のようなコードにしましたが先生のイメージと違いましたら教えていただけますと助かります。

#フォルダ名のリスト
d_namaes=os.listdir('./ocr/invoice3')
mylist=[]
for d_namae in d_namaes:
mylist.append(d_namae) #フォルダ名のリスト

#ファイル名の前半部分をとる
files=os.listdir('./ocr/invoice2')
for file in files:
file_name=os.path.splitext(os.path.basename(file))
file_name=file_name[0]
maeno_name=file_name[:4]

#リストから文字列を含むものを探す
for j in mylist:
if maeno_name in j:
path1=f'./ocr/invoice2/{file_name}.pdf'
path2=f'./ocr/invoice3/{j}/{file_name}.pdf'
os.rename(path1,path2)