python loc函数取出的数据类型_python – 使用pandas数据帧的.loc函数

您可以使用

between获取布尔值,然后使用astype将布尔值转换为0/1：

dtaframe['b'] = dtaframe['a'].between(0, 5, inclusive=False).astype(int)

结果输出：

a b

0 1 1

1 0 0

2 1 1

3 0 0

4 1 1

5 3 1

6 4 1

7 6 0

8 4 1

9 6 0

编辑

对于多个范围,您可以使用pandas.cut：

dtaframe['b'] = pd.cut(dtaframe['a'], bins=[0,1,6,9], labels=False, include_lowest=True)

您需要注意如何定义垃圾箱.使用labels = False将返回每个bin的整数指示符,这恰好与您提供的标签相对应.您也可以手动指定每个bin的标签,例如labels = [0,1,2],labels = [0,17,19],labels = [‘a’,’b’,’c’]等.如果手动指定标签,可能需要使用astype ,因为它们将作为类别返回.

或者,您可以组合loc和between来手动指定每个范围：

dtaframe.loc[dtaframe['a'].between(0,1), 'b'] = 0

dtaframe.loc[dtaframe['a'].between(2,6), 'b'] = 1

dtaframe.loc[dtaframe['a'].between(7,9), 'b'] = 2