フルパスに含まれるディレクトリ名を取得したい

Question

下のコードにあるパスの photo や cloud の文字列のみを取得したいです。
一つ上の階層までは取得できたのですが、2つ3つ上となると取得方法がわからなくなってしまいました。
良い方法がございましたら教えていただきたいです。

試したコード

import os

JpgFile = r'D:\python\photo\cloud\jpg\test.jpg'

dirname = os.path.dirname(JpgFile)
r = os.path.basename(dirname)
print(r)
# これだとjpgまでは取得できます

Answer 1

Python3.4以降をご利用なら、pathlibが便利です。

>>> from pathlib import Path
>>>
>>> jpg_path = Path(r'D:\python\photo\cloud\jpg\test.jpg')
>>> jpg_path.parts
('D:\', 'python', 'photo', 'cloud', 'jpg', 'test.jpg')
>>>
>>> jpg_path.parts[-2]
'jpg'
>>> jpg_path.parts[-3]
'cloud'
>>> jpg_path.parts[-4]
'photo'
>>>
>>> jpg_path.parts[-4:-1]
('photo', 'cloud', 'jpg')

Comments

[shamimatsu]

回答ありがとうございます！[]でインデックス指定して取り出せて便利ですね。

[LouiS0616]

pathlibはとにかく高機能で、os.pathのみならずshutilやglobなどの機能をも内包しています。
3.4以降ならば使わない手は無いです。

Answer 2

dirname = os.path.dirname(JpgFile)
dirname = os.path.dirname(dirname)
r = os.path.basename(dirname)

こーだった

Comments

[shamimatsu]

回答ありがとうございます！取得できました。