如何根据列表选择数据帧的一部分？[复制]

Question

我有以下数据框：

      import pandas as pd

      df = pd.DataFrame({'Name': ['MEXICO', 'CANADA', 'CANADA', 'PORTUGAL', 'ESPANHA', 
                                  'BRASIL', 'BRASIL', 'MEXICO'],                   
                         'Column_two': [1,2,3,4,5,6,7,8]                  
                         })

      print(df)

      # Output:

                  Name      Column_two
                 MEXICO        1
                 CANADA        2
                 CANADA        3
                PORTUGAL       4
                 ESPANHA       5
                  BRASIL       6
                  BRASIL       7
                  MEXICO       8

我有以下一些位于美国的城市列表：

      list_americ = ['MEXICO', 'CANADA', 'BRASIL']

我想在数据框中只保留列表\u americ中国家的“名称”。因此，我尝试执行以下代码：

     for correct_name in list_americ:
          df['Name'] = df['Name'].apply(lambda x: correct_name if correct_name == df['Name'] 
                       else x)

此代码产生以下错误：

      ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

我希望输出为：

                  Name      Column_two
                 MEXICO        1
                 CANADA        2
                 CANADA        3
                  BRASIL       6
                  BRASIL       7
                  MEXICO       8

Answer 1

使用df。loc（位置）：

df.loc[df['Name'].isin(list_americ)]