如果给定元素存在于子列表中，则Python返回该子列表的元素

Question

如何使python返回以下内容：

list = [['a', 23, 'h401'], ['f', 45, 'h403'], ['g', 56, 'h401']]

如果输入'h401' ，则应返回带有'h401'的子列表的索引[1]中'h401'

>>> 79                 #(23 + 56)

与给定'h403 '相同，它应返回'h403'为45的子列表的[1]

Answer 1

仅当第三个元素为h401并对它们求和时，才可以使用表达式选择第二个元素。

l = [['a', 23, 'h401'], ['f', 45, 'h403'], ['g', 56, 'h401']]

sum(i[1] for i in l if i[2] == 'h401')

如果只想检查目标元素是否在列表中，则可以使用：

sum(i[1] for i in l if 'h401' in i)

Answer 2

inp = input("input code")
mlist = [['a', 23, 'h401'], ['f', 45, 'h403'], ['g', 56, 'h401']]
count = 0
for i in range (len(mlist)):
    if (mlist[i][2] == inp):
        count += mlist[i][1]

print(count)