Python教程:拆分多級目錄的方法

Python探索牛發表於2024-05-21

實現多級目錄差分,舉例說明如下:

假設現有的目錄結構如下:1、2、2.1、2.2、2.3、2.4、3、4、5、6、6.1、6.1.1、6.1.2、6.1.3、6.1.4、6.2、6.3、6.4、7、8、9、10。

經過差分後,得到的各級目錄為:

  • 一級目錄:1、2、3、4、5、6、7、8、9、10

  • 二級目錄:2.1、2.2、2.3、2.4和6.1、6.2、6.3、6.4

  • 三級目錄:6.1.1、6.1.2、6.1.3、6.1.4

各級目錄的名字假設以字串的方式存放,且最後拆分出的各級目錄單獨存放。

實現原理

假設目錄等級最多三級,那麼我們如何實現呢?

我們可以透過給各級目錄賦一個權重,然後計算出各個目錄對應的權重值,進行比較之後,就可以得到各級目錄的分類。

假設各級目錄最多有1000個。

(1)一級目錄的權重就為1,則一級目錄的1到1000對應的權重就是1到1000。

1 ~ 1000 ==> 1 ~ 1000

一級目錄所佔的數值範圍:1 ~ 1000

(2)二級目錄的權重賦值如下:

1.1 ~ 1.100 ==> 1*1000+1 ~ 1*1000+100

2.1 ~ 2.100 ==> 2*1000+1 ~ 2*1000+100

'''''' ''''''

100.1 ~ 100.100 ==> 100*1000+1 ~ 100*1000+100

1000.1 ~ 1000.1000 ==> 1000*1000+1 ~ 1000*1000 + 1000

二級目錄所佔的數值範圍:1000 ~ 1001000

(3) 三級目錄的權重賦值如下:

1.1.1 ~ 1.1.100 ==> 1*1000000+1*1000+1 ~ 1*1000000+1*1000+100

1.1000.1 ~ 1.1000.1000 ==> 1*1000000+1000*1000+1~ 1*1000000+1000*1000+1000

1.2.1 ~ 1.2.1000 ==> 1*1000000+2*1000+1 ~ 1*1000000+2*1000+1000

最後,我們只需要將目錄按照上述的公式計算出權重,就可以根據權重獲得目錄所在的目錄結構是什麼。

程式碼實現

測試資料:

["1", "2", "2.1", "2.2", "2.3", "2.4", "3", "4", "5", "6", "6.1", "6.1.1", "6.1.2", "6.1.3", "6.1.4", "6.2", "6.3", "6.4", "7", "8", "9", "10"]
# encoding: utf-8
import os
import sys
import logging
#學習中遇到問題沒人解答?小編建立了一個Python學習交流群:153708845
# 設定logging.basicConfig()方法的引數和配置logging.basicConfig函式
FORMAT = '[%(funcName)s: %(lineno)d]: %(message)s'
LEVEL = logging.INFO
logging.basicConfig(level = LEVEL, format=FORMAT)


'''
* 獲取二級標題所在目錄名
* 返回值:二級目錄標題名
目錄與數值對應關係:
1.1 ~ 1.100 ==>  1*1000+1 ~ 1*1000+100
2.1 ~ 2.100 ==>  2*1000+1 ~ 2*1000+100
'''
def judge_second_directory(get_num):
    for i in range(1, 1001, 1):
        if get_num > i * 1000 and get_num < (i+1)*1000:
            return str(i) + ".1"
    return None


'''
* 獲取三級標題所在目錄名
* 返回值:三級目錄標題名
目錄與數值對應關係:
1.1.1 ~ 1.1.100  ==> 1*1000000+1*1000+1 ~ 1*1000000+1*1000+100
1.2.1 ~ 1.2.1000 ==> 1*1000000+2*1000+1 ~ 1*1000000+2*1000+1000
'''
def judge_three_directory(get_num):
    ret_str = None
    for i in range(1, 1001, 1):
        # 判斷二級標題
        if get_num > i * 1000000 and get_num < (i+1)*1000000:
            ret_str = str(i) + "."
        for j in range(1, 1001, 1):
            # 判斷三級標題
            if get_num > i*1000000 + j*1000 and get_num < i*1000000 + (j+1)*1000:
                ret_str += str(j) + '.1'
                return ret_str
    return None


'''
按目錄進行分類:
分類原理:
(1) 一級目錄:
1 ~ 1000     ==>  1 ~ 1000
一級目錄所佔的數字範圍:1 ~ 1000
(2) 二級目錄:
1.1 ~ 1.100 ==>  1*1000+1 ~ 1*1000+100
2.1 ~ 2.100 ==>  2*1000+1 ~ 2*1000+100
''''''
''''''
100.1 ~ 100.100    ==> 100*1000+1 ~ 100*1000+100
1000.1 ~ 1000.1000 ==> 1000*1000+1 ~ 1000*1000 + 1000
二級目錄所佔的數字範圍:1000 ~ 1001000
(3) 三級目錄:
1.1.1 ~ 1.1.100        ==> 1*1000000+1*1000+1 ~ 1*1000000+1*1000+100
1.1000.1 ~ 1.1000.1000 ==> 1*1000000+1000*1000+1000
1.2.1 ~ 1.2.1000 ==> 1*1000000+2*1000+1 ~ 1*1000000+2*1000+1000
'''
def dirctory_classify(directory_list):
    parent_directory = []
    second_directory = {}
    three_directory  = {}
    for directory_item in directory_list:
        num_list = directory_item.split(".")
        # 開始按目錄進行分類
        if len(num_list) == 1:   # 一級目錄
            parent_directory.append(directory_item)
        elif len(num_list) == 2: # 二級目錄
            # 例:1.1 ==> 1*1000+1
            get_num = int(num_list[0]) * 1000 + int(num_list[1])
            dir_tmp = judge_second_directory(get_num)
            if dir_tmp == None:
                continue
            if dir_tmp not in second_directory:
                second_directory[dir_tmp] = [directory_item]
            else:
                tmp_list = second_directory[dir_tmp]
                tmp_list.append(directory_item)
                second_directory[dir_tmp] = tmp_list
        elif len(num_list) == 3: # 三級目錄
            # 例:1.1.100 ==> 1*1000000+1*1000+100
            get_num = int(num_list[0])*1000000 + int(num_list[1])*1000 + int(num_list[2])
            dir_tmp = judge_three_directory(get_num)
            if dir_tmp == None:
                continue
            if dir_tmp not in three_directory:
                three_directory[dir_tmp] = [directory_item]
            else:
                tmp_list = three_directory[dir_tmp]
                tmp_list.append(directory_item)
                three_directory[dir_tmp] = tmp_list
        else:
            logging.error("這是一個超過三級目錄的條目!!!!")
            logging.error("num: %s" % directory_item)
    return [parent_directory, second_directory, three_directory]


if __name__ == '__main__':
    original_data = ["1", "2", "2.1", "2.2", "2.3", "2.4", "3", "4", "5", "6", "6.1", "6.1.1", "6.1.2", "6.1.3", "6.1.4", "6.2", "6.3", "6.4", "7", "8", "9", "10"]
    directory = dirctory_classify(original_data)

    parent_directory = directory[0]
    second_directory = directory[1]
    three_directory  = directory[2]

    print ("一級目錄:", parent_directory)
    print ("二級目錄:", second_directory)
    print ("三級目錄:", three_directory)

執行測試:
在這裡插入圖片描述
可以看出,測試結果符合預期,一共有三級目錄。各級目錄中的目錄結構也劃分了出來。

相關文章