Prepare tree diagram data from cluster file list

test.py

import os
import glob

# get filename list
list = [os.path.basename(f) for f in glob.glob('./test/*')]
#['1-2.csv', '1-1.csv', '0.csv', '1-2-2.csv', '1.csv', '1-2-1.csv', '2.csv']

# remove 0.csv and extension
list.remove('0.csv')
list = [x.replace('.csv', '') for x in list]
#['1-2', '1-1', '1-2-2', '1', '1-2-1', '2']

new_list = []
level = 3

for x in list:
    # split filename to sub list
    temp = [int(i) for i in (x.split("-"))]
    #[[1, 2], [1, 1], [1, 2, 2], [1], [1, 2, 1], [2]]

    # fill empty cell with 0
    while len(temp) < level:
        temp.append(0)
    new_list.append(temp)
    #[[1, 2, 0], [1, 1, 0], [1, 2, 2], [1, 0, 0], [1, 2, 1], [2, 0, 0]]

new_list.sort()
#[[1, 0, 0], [1, 1, 0], [1, 2, 0], [1, 2, 1], [1, 2, 2], [2, 0, 0]]