python pandas loc 布尔索引示例说明

  • Post category:Python

下面就为大家详细讲解“python pandas loc 布尔索引示例说明”的完整攻略。

1. loc的基本用法

1.1 loc的语法

df.loc[行标签,列标签]

loc的第一个参数是行标签,第二个参数是列标签。行标签包括index和boolean array,列标签是包括column name和boolean array。

1.2 示例说明

  • 在之前导入pandas库的基础上,首先需要创建一个dataframe类型的数据
import pandas as pd
df=pd.DataFrame({'A':[1,2,3],'B':[4,5,6],'C':[7,8,9]})
  • 通过loc方法取数据
df.loc[1,'B']

以上代码的含义是取出df数据中第二行,B列的数据,输出结果为5.

2. loc的布尔索引用法

2.1 布尔索引的语法

df.loc[boolean1,boolean2]

其中boolean1和boolean2是相应的True和False的boolean array。

2.2 示例说明

  • 在之前创建的df数据中,现在要取出所有A列大于等于2的数据
df.loc[df['A'] >= 2,:]

以上代码的含义是取出df数据中A列大于等于2的所有数据,输出结果为:

    A   B   C
1   2   5   8
2   3   6   9
  • 再举一个例子,现在要取出所有A列小于2,B列等于6的数据
df.loc[(df['A'] < 2) & (df['B'] == 6), :]

以上代码的含义是取出df数据中A列小于2,B列等于6的所有数据,输出结果为空值,因为该条件没有符合的数据。

至此,我们已经完成了“python pandas loc 布尔索引示例说明”的完整攻略,相信大家已经对该主题有了一个更加深入的理解。