import argparse def main(): parser = argparse.ArgumentParser() parser.add_argument('--fileNameList', type=str) args = parser.parse_args() name_list = args.fileNameList.split(":") folder_need_check = set() for loc in name_list: # Find only the sub-folder of 'example' folder if loc.split("/")[0] == "examples" and len(loc.split("/")) >= 4: folder_need_check.add(loc.split("/")[1] + "/" + loc.split("/")[2]) # Output the result using print. Then the shell can get the values. print(list(folder_need_check)) if __name__ == '__main__': main()