📜  数据框选择列表中的条目 - Python (1)

📅  最后修改于: 2023-12-03 14:54:55.645000             🧑  作者: Mango

数据框选择列表中的条目 - Python

要在Python中选择一个数据框(DataFrame)中列表(list)中的特定条目,可以使用以下方法:

import pandas as pd

# 创建一个数据框Dataframe
df = pd.DataFrame({'Name': ['Alice', 'Bob', 'Charlie'], 'Age': [25, 30, 35], 'City': ['Beijing', 'Shanghai', 'Guangzhou']})

# 创建一个列表List
cities_to_check = ['Beijing', 'Shanghai']

# 使用isin()方法选择出数据框中匹配到的条目
result_df = df[df['City'].isin(cities_to_check)]

# 在终端打印出结果
print(result_df)

这个程序会返回一个包含符合条件的数据框(DataFrame),内容是:

    Name  Age      City
0  Alice   25   Beijing
1    Bob   30  Shanghai

其中isin()函数用来检查是否为列表提供的几个值之一的函数。在我们的例子中,我们传递了一个需要检查的城市列表cities_to_check到这个函数中,这个函数会返回一个布尔型数据帧,表示每个行是否包含列表中包含的元素。最后,我们使用df[]选择包含True的行,以选择符合条件的行。这个程序在这个列表中选择了出现在数据框中的北京和上海。

以上就是用Python在数据框中选择列表中特定条目的方法。

参考资料:

  • Pandas documentation: https://pandas.pydata.org/docs/user_guide/index.html
  • isin()函数的使用方法: https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.isin.html