检查来自两个 arrays 的元素

Question

我有两个不同的 arrays，我需要知道第一个数组“date”中的名称是否在第二个数组“name”中

date = ['Olivia Smith', 'Jacob Williams', 'Jessica Taylor', 'Harry Miller']
name = ['Olivia', 'Jessica', 'Jacob', 'James']

for i in range(len(data)):
    x = 0
    print(name.index(data[x]))
    x = x + 1
    i = i + 1

我希望程序确认“数据”数组中的名称在“名称”数组中

Answer 1

使用列表理解和string.split()只获取名字。 然后使用另一个列表理解来获取相交。

date = ['Olivia Smith', 'Jacob Williams', 'Jessica Taylor', 'Harry Miller']
name = ['Olivia', 'Jessica', 'Jacob', 'James']

date_fn = [full_name.split()[0] for full_name in date]

names_in_both = [first_name for first_name in date_fn if first_name in name]

Answer 2

使用 split 并抓住第一部分。 然后迭代比较。

new_date = [name.split(' ')[0] for name in date]

for name in set(name):
    if name in new_date:
        print(f"{name} is found!.")

Answer 3

您可以使用理解：

out = [sum(x in y for y in date) for x in name]
print(*zip(name, out), sep='\n')

# Ouput:
('Olivia', 1)
('Jessica', 1)
('Jacob', 1)
('James', 0)

Answer 4

如果你想检查是否存在任何一个（返回 bool）：

any(i.split()[0] in name for i in date)

如果你想检查是否全部存在（返回 bool）：

any(i.split()[0] in name for i in date)

如果你想检索存在的那些：

def intersection(a, b):
    res = []
    for i in a:
        if i.split()[0] in b:
            res.append(i.split()[0])
    return res
print(intersection(date, name))
# Output: ['Olivia', 'Jacob', 'Jessica']

可以将上面的function简化为：

def intersection(a, b):
    res = [i.split()[0] for i in a if i.split()[0] in b]
    return res

文档：
any()
all()
列表理解