在多维列表的特定位置插入元素

Question

我正在尝试在多维列表的特定实例中插入一个元素：

eg_list = [[['DF1 IN Element', ['DF_CAT1_APPLE_SELECT', 'Element_TO()', '1']], ['DOG Element', ['DF1_CAT_en', 1.0, 'DF_CAT1_DOG', 'Element_TO()', '1']], ['BANANA Element', ['DF_CAT1_BANANA', 'Element_TO()', '1']]], [['DF2 IN Element', ['DF_CAT2_APPLE_SELECT', 'Element_TO()', '1']], ['DOG Element', ['DF_CAT2_DOG', 'Element_TO()', '1']], ['BANANA Element', ['DF_CAT2_BANANA', 'Element_TO()', '1']]], [['MB1 IN ENABLE', ['MB_CAT1_APPLE_SELECT', 'Element_TO()', '1']], ['DOG Element', ['MB_CAT1_DOG', 'Element_TO()', '1']], ['BANANA Element', ['MB_CAT1_BANANA', 'Element_TO()', '1']]], [['MB2 IN ENABLE', ['MB_CAT2_APPLE_SELECT', 'Element_TO()', '1']], ['DOG Element', ['MB_CAT2_DOG', 'Element_TO()', '1']], ['BANANA Element', ['MB_CAT2_BANANA', 'Element_TO()', '1']]], [['ABC1 IN ENABLE', ['ABC_CAT1_APPLE_SELECT', 'Element_TO()', '1']], ['DOG Element', ['ABC_CAT1_DOG', 'Element_TO()', '1']], ['BANANA Element', ['ABC_CAT1_BANANA', 'Element_TO()', '1']]]]

for i in range(0, len(eg_list)):
    for j in range(0, len(eg_list[i])):
        for k in range(0, len(eg_list[i][j][1])):
            if eg_list[i][j][1][k] == 'Element_TO()':
                eg_list.insert([i][j][1][k], 'MAIN')

基本上，我想要做的是，如果它在列表中找到“Element_TO()”，那么我会尝试在该元素之后插入“MAIN”。我正在寻找的输出是 -

[[['DF1 IN Element', ['DF_CAT1_APPLE_SELECT', 'Element_TO()', 'MAIN', '1']], ['DOG Element', ['DF1_CAT_en', 1.0, 'DF_CAT1_DOG', 'Element_TO()', 'MAIN', '1']], ['BANANA Element', ['DF_CAT1_BANANA', 'Element_TO()', 'MAIN', '1']]], [['DF2 IN Element', ['DF_CAT2_APPLE_SELECT', 'Element_TO()', 'MAIN', '1']], ['DOG Element', ['DF_CAT2_DOG', 'Element_TO()', 'MAIN', '1']], ['BANANA Element', ['DF_CAT2_BANANA', 'Element_TO()', 'MAIN', '1']]], [['MB1 IN ENABLE', ['MB_CAT1_APPLE_SELECT', 'Element_TO()', 'MAIN', '1']], ['DOG Element', ['MB_CAT1_DOG', 'Element_TO()', 'MAIN', '1']], ['BANANA Element', ['MB_CAT1_BANANA', 'Element_TO()', 'MAIN', '1']]], [['MB2 IN ENABLE', ['MB_CAT2_APPLE_SELECT', 'Element_TO()', 'MAIN', '1']], ['DOG Element', ['MB_CAT2_DOG', 'Element_TO()', 'MAIN', '1']], ['BANANA Element', ['MB_CAT2_BANANA', 'Element_TO()', 'MAIN', '1']]], [['ABC1 IN ENABLE', ['ABC_CAT1_APPLE_SELECT', 'Element_TO()', 'MAIN', '1']], ['DOG Element', ['ABC_CAT1_DOG', 'Element_TO()', 'MAIN', '1']], ['BANANA Element', ['ABC_CAT1_BANANA', 'Element_TO()', 'MAIN', '1']]]]

我怎样才能得到这个？

编辑 - 发布了另一个包含此问题内容的问题，以获得更通用的解决方案。链接到那个问题 - link to the question

Answer 1

我会为任务使用递归：

def add(l):
    if isinstance(l, list):
        if "Element_TO()" in l:
            l.insert(l.index("Element_TO()") + 1, "MAIN")
        for v in l:
            add(v)


add(eg_list)
print(eg_list)

打印：

[
    [
        [
            "DF1 IN Element",
            ["DF_CAT1_APPLE_SELECT", "Element_TO()", "MAIN", "1"],
        ],
        [
            "DOG Element",
            ["DF1_CAT_en", 1.0, "DF_CAT1_DOG", "Element_TO()", "MAIN", "1"],
        ],
        ["BANANA Element", ["DF_CAT1_BANANA", "Element_TO()", "MAIN", "1"]],
    ],
    [
        [
            "DF2 IN Element",
            ["DF_CAT2_APPLE_SELECT", "Element_TO()", "MAIN", "1"],
        ],
        ["DOG Element", ["DF_CAT2_DOG", "Element_TO()", "MAIN", "1"]],
        ["BANANA Element", ["DF_CAT2_BANANA", "Element_TO()", "MAIN", "1"]],
    ],
    [
        [
            "MB1 IN ENABLE",
            ["MB_CAT1_APPLE_SELECT", "Element_TO()", "MAIN", "1"],
        ],
        ["DOG Element", ["MB_CAT1_DOG", "Element_TO()", "MAIN", "1"]],
        ["BANANA Element", ["MB_CAT1_BANANA", "Element_TO()", "MAIN", "1"]],
    ],
    [
        [
            "MB2 IN ENABLE",
            ["MB_CAT2_APPLE_SELECT", "Element_TO()", "MAIN", "1"],
        ],
        ["DOG Element", ["MB_CAT2_DOG", "Element_TO()", "MAIN", "1"]],
        ["BANANA Element", ["MB_CAT2_BANANA", "Element_TO()", "MAIN", "1"]],
    ],
    [
        [
            "ABC1 IN ENABLE",
            ["ABC_CAT1_APPLE_SELECT", "Element_TO()", "MAIN", "1"],
        ],
        ["DOG Element", ["ABC_CAT1_DOG", "Element_TO()", "MAIN", "1"]],
        ["BANANA Element", ["ABC_CAT1_BANANA", "Element_TO()", "MAIN", "1"]],
    ],
]