python在dataframe中查找特定值

import pandas as pd

# 创建一个示例DataFrame
data = {
    'A': [1, 2, 3],
    'B': ['foo', 'bar', 'baz']
}
df = pd.DataFrame(data)

# 使用loc查找特定值
value = df.loc[df['A'] == 2, 'B'].values[0]
print(value)  # 输出: bar

# 使用query查找特定值
value = df.query("A == 2")['B'].values[0]
print(value)  # 输出: bar

# 使用isin查找特定值
values_to_find = [2, 3]
filtered_df = df[df['A'].isin(values_to_find)]
print(filtered_df)  # 输出:    A    B\n1  2  bar\n2  3  baz\n"

# 使用apply查找特定值
def find_value(row):
    if row['A'] == 2 and row['B'] == 'bar':
        return True
    return False

result = df.apply(find_value, axis=1)
print(result)  # 输出: 0    False\n1     True\n2    False\ndtype: bool"