📅  最后修改于: 2023-12-03 14:54:55.645000             🧑  作者: Mango
要在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在数据框中选择列表中特定条目的方法。
参考资料:
isin()
函数的使用方法: https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.isin.html