循环遍历两个列表不起作用

Question

我正在尝试使用循环来遍历两个列表。不幸的是，第二个 for 循环不起作用：它只检查列表中的第一项，而不是其余的。 你能告诉我为什么吗？

谢谢

列表：

low_cars_engines=['Audi', 'Bentley', 'Bugatti', 'Porsche', 'Skoda']
low_planes_engines=['Pratt & Whitney','Rolls-Royce','GE Aviation']

我想根据 if 语句在我的原始数据集中再添加两列（汽车和飞机）：

• 如果列表 'Engine to check' 中的对象在列表 low_cars_engines 中，则它是汽车，否则，它不是；
• 如果列表“Engine to check”中的对象在列表low_planes_engines 中，则它是平面，否则不是。

import re

df['Cars'] = pd.Series(index = df.index, dtype='object')
df['Planes'] = pd.Series(index = df.index, dtype='object')

for index, row in df.iterrows():
    value = row['Engine to check']
    for x in low_cars_engines:
        if x in value:
            print(x)
            df.at[index,'Cars'] = 'Yes' # need to keep df.at[index, '_']
            break
        else: 
            df.at[index,'Cars'] = 'No' # need to keep df.at[index, '_']
            break

for index, row in df.iterrows():
    value = row['Engine to check']
    for x in low_planes_engines:
        if x in value:
            df.at[index,'Planes'] = 'Yes'
            break
        else: 
            df[index,'Planes'] = 'No'
            break

print(df)

第一个 for 循环工作正常，但不是第二个：我无法为列表“Engine to check”中的项目分配值，即使它在列表 low_planes_engines 中（它总是给我否）。

您能告诉我哪里出了问题吗？是否可以只使用一个 for 循环而不是两个？我宁愿保持相同的结构，或者保持df.at[index,'_']。现在，第二个循环仅打印/检查列表 low_planes_engines 的第一项（即 Pratt & Whitney），其余的不进行。

由于数据集类似于：

Audi
CFM International
Rolls-Royce
Bentley
Volkswagen
Toyota
Suzuki
Porsche

并且它不包括该元素，Planes 下的所有行都设置为No。

Answer 1

你这里有一个额外的空间

row['Engine to check ']

试试改成

row['Engine to check']

Answer 2

在使用 Pandas 时不应使用循环。 DataFrames 不是为顺序访问而设计的。不过，你需要一些 NumPy：

import numpy as np
df['Cars']   = np.where(df['Engine to check'].isin(low_cars_engines), 'Yes', 'No') 
df['Planes'] = np.where(df['Engine to check'].isin(low_planes_engines), 'Yes', 'No')

结果：

#     Engine to check Cars Planes
# 0               Audi  Yes     No
# 1  CFM International   No     No
# 2        Rolls-Royce   No    Yes
# 3            Bentley  Yes     No
# 4         Volkswagen   No     No
# 5             Toyota   No     No
# 6             Suzuki   No     No
# 7            Porsche  Yes     No

您可能也不应该使用“是”和“否”。请改用布尔值 True 和 False，因为它们将来更易于使用：

df['Cars']   = df['Engine to check'].isin(low_cars_engines) 
df['Planes'] = df['Engine to check'].isin(low_planes_engines)

最后，如果 DataFrame 中的所有内容严格来说都是汽车或飞机，则只需要一列。另一个是补语。