Python:从嵌套的字典列表中删除项目，跳过元素

Question

我在从嵌套列表中删除项目时遇到问题，似乎遇到了障碍。

我有许多测试目录，每个目录都有特定的测试模式。如果参数 KEEP_DIR 设置为 True，我需要目录。如果不是，我需要从我的列表中删除 test_mode。

我试图创建一个单独的我不需要的测试模式列表，然后删除它们，而不对原始列表进行索引，但是一旦从 dir_dict 中删除索引，我的代码就会跳过列表项。

我知道列表理解可能是最好的方法，但我看到的每个示例都是针对一维列表的，我一直在努力理解如何将我看到的示例改编成我的代码。

这是我的嵌套列表:

{
    "TEST_DIRS": [
        {
            "BASE_DIR": "C:\\Path\\to\\files",
            "TEST_MODES": [
                {
                    "DIRS": {
                        "DIR_1": "C:\\Path\\to\\files\\tests\\",
                        "DIR_2": "C:\\Path\to\\files\\logs\\"
                    },
                    "FILES": {
                        "FILE_1": "C:\\Path\\to\\files\\tests\\file1",
                        "FILE_2": "C:\\Path\\to\\files\\tests\\file2"
                    },
                    "KEEP_DIR": true
                },
                {
                    "DIRS": {
                        "DIR_1": "C:\\another\\path\\to\\files\\tests\\",
                        "DIR_2": "C:\\another\\path\\to\\files\\log\\"
                    },
                    "FILES": {
                        "FILE_1": "C:\\another\\path\\to\\files\\tests\\file1",
                        "FILE_2": "C:\\another\\path\\to\\files\\tests\\file2"
                    },
                    "KEEP_DIR": false
                }
            ]
        }
    ]
}

这是我使用的代码:

for i, base_dir in enumerate(dir_dict['TEST_DIRS']):
    for n, test_mode in enumerate(base_dir['TEST_MODES']):
        if not test_mode['KEEP_DIR']:
            fail = (i, n)
            del_list.append(fail)

for item in del_list:
    try:
        del dir_dict['TEST_DIRS'][item[0]]['TEST_MODES'][item[1]]
    except:
        print("Failed to delete:", item)

输出:

Deleting: (0, 0)
Deleting: (0, 1)
Deleting: (0, 2)
Deleting: (0, 3)
Deleting: (0, 4)
Deleting: (0, 5)
Deleting: (0, 6)
Deleting: (0, 7)
Deleting: (0, 8)
Deleting: (0, 9)
Failed to delete: (0, 9)
Deleting: (0, 10)
Failed to delete: (0, 10)
Deleting: (0, 12)
Failed to delete: (0, 12)
Deleting: (0, 13)
Failed to delete: (0, 13)
Deleting: (0, 14)
Failed to delete: (0, 14)
Deleting: (0, 15)
Failed to delete: (0, 15)
Deleting: (0, 16)
Failed to delete: (0, 16)

回溯:当我删除 try, except block 时 - 这是我得到的错误:

Traceback (most recent call last):
:
:
  File "check_files.py", line 180, in check_dir_dict
    del dir_dict['TEST_DIRS'][item[0]]['TEST_MODES'][item[1]]
IndexError: list assignment index out of range

非常感谢任何帮助。

解决方案:正如 ZZ ll 所指出的，反转 del_list 应该可以消除索引问题。当我颠倒 del_list 的顺序时，索引仍然被删除，但它们不会在每次删除之间移动。这看起来是目前最容易实现的解决方案:

for item in reversed(del_list):
    del dir_dict['SAGE_DIRS'][item[0]]['TEST_MODES'][item[1]]

感谢所有的帮助:)

Answer 1

此处将递归遍历您的数据结构并删除所有包含 'KEEP_DIR' 键且关联值为 'false' 的字典:

def filter_out(orig):
  if isinstance(orig, list):
    return [ filter_out(element)
             for element in orig
             if (not isinstance(element, dict) or
                 element.get('KEEP_DIR', 'true') != 'false') ]
  elif isinstance(orig, dict):
    return { key: filter_out(value)
             for key, value in orig.items() }
  else:
    return orig

您可能需要调整为使用 False 而不是 'false' 或类似的，具体取决于您的实际值。 false 在 Python 中无效。

dir_dict = {
    "TEST_DIRS": [
        {
            "BASE_DIR": "C:\\Path\\to\\files",
            "TEST_MODES": [
                {
                    "DIRS": {
                        "DIR_1": "C:\\Path\\to\\files\\tests\\",
                        "DIR_2": "C:\\Path\\to\\files\\logs\\",
                    },
                    "FILES": {
                        "FILE_1": "C:\\Path\\to\\files\\tests\\file1",
                        "FILE_2": "C:\\Path\\to\\files\\tests\\file2",
                    },
                    "KEEP_DIR": 'true',
                },
                {
                    "DIRS": {
                        "DIR_1": "C:\\another\\path\\to\\files\\tests\\",
                        "DIR_2": "C:\\another\\path\\to\\files\\log\\",
                    },
                    "FILES": {
                        "FILE_1": "C:\\another\\path\\to\\files\\tests\\file1",
                        "FILE_2": "C:\\another\\path\\to\\files\\tests\\file2",
                    },
                    "KEEP_DIR": 'false',
                },
           ]
        }
    ]
}

现在你可以这样调用它:

import pprint
pprint.pprint(dir_dict)
{'TEST_DIRS': [{'BASE_DIR': 'C:\\Path\\to\\files',
                'TEST_MODES': [{'DIRS': {'DIR_1': 'C:\\Path\\to\\files\\tests\\',
                                         'DIR_2': 'C:\\Path\\to\\files\\logs\\'},
                                'FILES': {'FILE_1': 'C:\\Path\\to\\files\\tests\\file1',
                                          'FILE_2': 'C:\\Path\\to\\files\\tests\\file2'},
                                'KEEP_DIR': 'true'},
                               {'DIRS': {'DIR_1': 'C:\\another\\path\\to\\files\\tests\\',
                                         'DIR_2': 'C:\\another\\path\\to\\files\\log\\'},
                                'FILES': {'FILE_1': 'C:\\another\\path\\to\\files\\tests\\file1',
                                          'FILE_2': 'C:\\another\\path\\to\\files\\tests\\file2'},
                                'KEEP_DIR': 'false'}]}]}
pprint.pprint(filter_out(dir_dict))
{'TEST_DIRS': [{'BASE_DIR': 'C:\\Path\\to\\files',
                'TEST_MODES': [{'DIRS': {'DIR_1': 'C:\\Path\\to\\files\\tests\\',
                                         'DIR_2': 'C:\\Path\\to\\files\\logs\\'},
                                'FILES': {'FILE_1': 'C:\\Path\\to\\files\\tests\\file1',
                                          'FILE_2': 'C:\\Path\\to\\files\\tests\\file2'},
                                'KEEP_DIR': 'true'}]}]}