python - 创建包含特定元素的数组列表

Question

我需要创建一个函数来搜索某个字符串是否在数组中，并创建一个包含包含它的所有元素(列表)的列表。例如:

word = 'busy'
array = [[['99', 'Normal', [], ['busy', '0'], 28016.2, 'working', '1']],[['F27', 'Normal', [], ['free', '0'], 28016.537865230806, 'working', '1']]]

所以我的输出应该是:

[['99', 'Normal', [], ['busy', '0'], 362.01, 'working', '1']]

但是当它显然存在时，我只得到表明字符串不存在的验证。这是代码:

array = [[['99', 'Normal', [], ['busy', '0'], 28016.2, 'working', '1']],[['F27', 'Normal', [], ['free', '0'], 28016.537865230806, 'working', '1']]]

def searchBusyWorkers(array):
    busy = []
    for x in array:
        if 'busy' in x:
            ind = x.index('busy')
            busy.append(array[ind])
            return busy
        else:
            return "No workers have that condition."

Answer 1

'busy' in x 失败，因为“busy”嵌套更深一层。换句话说，x 中没有 'busy'，但 x 中有一个包含字符串“busy”的子数组。

由于您提到“忙碌”将始终位于特定索引处，因此您需要解压缩以获取子数组，然后检查该索引 -

for x in array:
    elements, *_ = x
    if 'busy' in elements[3]:
        print(x)

输出

[['99', 'Normal', [], ['busy', '0'], 28016.2, 'working', '1']]