以下是一个示例代码,可以按照两列数据的条件进行子集取值:
import pandas as pd
# 创建示例数据
data = {'A': [1, 2, 3, 4, 5],
'B': [6, 7, 8, 9, 10],
'C': ['a', 'b', 'c', 'd', 'e']}
df = pd.DataFrame(data)
# 定义条件
condition1 = df['A'] > 2
condition2 = df['B'] < 9
# 使用条件进行子集取值
subset = df[condition1 & condition2]
# 输出结果
print(subset)
输出结果为:
A B C
2 3 8 c
上述代码中,我们首先创建了一个包含三个列的DataFrame对象。然后,我们定义了两个条件condition1和condition2,分别表示列A大于2和列B小于9的条件。最后,我们使用条件1和条件2进行逻辑与运算,并将结果赋值给subset变量,即按照两列数据的条件进行子集取值。最后,我们通过打印subset来查看结果。