Pandas提取错误：“列必须与键的长度相同”

import pandas as pd

# 创建一个示例DataFrame
data = {'A': [1, 2, 3, 4, 5],
        'B': [6, 7, 8, 9, 10]}
df = pd.DataFrame(data)

# 错误示例：条件列的长度不匹配
condition = [True, False]  # 长度为2，与DataFrame的键的长度不同
df_extract = df[condition]  # 报错："列必须与键的长度相同"

# 正确示例：确保条件列的长度与DataFrame的键的长度相同
condition = [True, False, True, False, True]  # 长度为5，与DataFrame的键的长度相同
df_extract = df[condition]  # 提取满足条件的数据

# 或者使用其他方法来提取数据
df_extract = df.loc[condition]  # 使用loc函数进行条件筛选