Question

我需要比较两个不同的数据是否匹配不同的列表。

我有这两个列表，我需要用：来计算婴儿的数量：

first_name_baby = S AND age_baby = 1 
age_baby = [ 2, 1, 3, 1, 4, 2, 4, 1, 1, 3, 4, 2, 2, 3].  first_name_baby= [ T, S, R, T, O, A, L, S, F, S, Z, U, S, P]

first_name_baby = S AND age_baby = 1时实际上有2次，但是我需要为此编写一个Python程序。

Answer 1

使用zip组合相应的列表条目，然后使用.count

>>> age_baby = [ 2, 1, 3, 1, 4, 2, 4, 1, 1, 3, 4, 2, 2, 3]
>>> first_name_baby = "T, S, R, T, O, A, L, S, F, S, Z, U, S, P".split(', ')
>>> list(zip(first_name_baby, age_baby)).count(('S', 1))
2

或者，您可以使用numpy。这将提供与您尝试过的解决方案非常相似的解决方案：

>>> import numpy as np
>>>                                                                                                             
>>> age_baby = np.array(age_baby)                                                    
>>> first_name_baby = np.array(first_name_baby)                                      
>>>                                                                                                                 
>>> np.count_nonzero((first_name_baby == 'S') & (age_baby == 1))                                      
2

Answer 2

只要条件匹配，您就可以取1的总和。使用zip同时遍历列表：

# need to make sense of the names
T, S, R, O, A, L, F, Z, U, S, P = 'T, S, R, O, A, L, F, Z, U, S, P'.split(', ')
age_baby = [2, 1, 3, 1, 4, 2, 4, 1, 1, 3, 4, 2, 2, 3]
first_name_baby = [T, S, R, T, O, A, L, S, F, S, Z, U, S, P]


sum(1 for age, name in zip(age_baby, first_name_baby) 
    if age == 1 and name == S)

感谢Austin的更为优雅的版本：

sum(age == 1 and name == S for age, name in zip(age_baby, first_name_baby))

之所以可行，是因为python中的布尔是int的子类，而True基本上是1（带有重载的__str__和__repr__）和{{1} }是False；因此可以将布尔值相加，结果就是0个比较的数量。

Answer 3

尝试一下：

>>> count = 0
>>> 
>>> 
>>> for i in range(len(first_name_baby)):
...   if first_name_baby[i] == 'S' and age_baby[i] == 1:
...     count += 1
... 
>>> count
2

Answer 4

x = len([item for idx, item in enumerate(age_baby) if item == 1 and first_name_baby[idx] == 'S'])

展开：

l = []
for idx, item in enumerate(age_baby):
    if item == 1 and first_name_baby[idx] == 'S':
        l.append(item)
x = len(l)

比较两个不同列表中的不同元素

4 个答案: