python - 根据每个元组内的值对元组列表进行分区

Question

我正在尝试将一组数据排序到两个单独的列表中，全职和兼职。但它似乎不起作用。有人能指出我哪里出错了吗？

data = [(['Andrew'], ['FullTime'], [38]), 
       (['Fred'], ['PartTime'], [24]), 
       (['Chris'], ['FullTime'], [38])]

def sort(var1, datadump):
    positionlist = []
    for b in range(0, len(datadump)):
        temp2 = datadump[b][1]
        if (temp2 == var1):
            positionlist.append(datadump[b])
    return (positionlist)

FullTimeList = sort("FullTime", data) 
PartTimeList = sort("PartTime", data)

print(FullTimeList) 
print(PartTimeList)

Answer 1

这可以通过改变来解决

if (temp2 == var1):

到

if (temp2[0] == var1):

这是因为每个元组中的元素都是包含字符串的列表，而不是字符串本身。

这个问题也可以使用两个列表理解来解决:

FullTimeList = [x for x in data if x[1][0] == 'FullTime']
PartTimeList = [x for x in data if x[1][0] == 'PartTime']