python - 将一个列表分成两个列表

Question

下面是我当前的训练列表；

Training_list = [37, 0.7173543689320389, None, None, 9, 0.3351132686084142, 0.05165857605177993, 0.2942961165048544, 0.8373381877022654, 0.6119741100323625, 0, 0, 45, None, ' <=50K'], [46, 0.7173543689320389, None, None, 13, 0.03673139158576052, 0.13199838187702265, 0.3013349514563107, 0.8373381877022654, 0.38802588996763754, 0, 0, 25, None, ' <=50K'], [44, 0.7173543689320389, None, None, 9, 0.1610032362459547, 0.12823624595469255, 0.3013349514563107, 0.8373381877022654, 0.6119741100323625, 0, 0, 40, None, ' >50K']

我需要根据最后一个属性“<=50K”或“>50K”将列表分为两个列表
我想从两个列表中排除最后一个属性。
我尝试了下面的代码，得到以下结果:

under 50k list
[]

代码如下；

over_50k = []
under_50k = []
for row in training_list:
        if row[-1].lstrip() == " <=50K":
            under_50k.append(row[:-1])
        elif row [-1].lstrip() == " >50K":
            over_50k.append(row[:-1])

print("under 50k list")
print(under_50k)

非常感谢任何帮助。

Answer 1

1. 您正在使用lstrip() .
2. lstrip()将删除前导空格。
3. 字符串" <=50K"有前导空格。

因此我们可能会尝试与没有前导空格的字符串进行比较:

if row[-1].strip() == "<=50K":

或者也许:

under_50k = [row[:-1] for row in training_list if row[-1].strip() == "<=50K"]
over_50k  = [row[:-1] for row in training_list if row[-1].strip() == ">50K"]