熊猫:按范围对列值进行分类

Question

我有一个数据帧,df是这样的

a  b  c
1  0  2
5  7  8
4  1  3
3  4  6
5  2  5
.......

现在我想用以下范围替换这些数据

0-3 = 1
4-6 = 2
7-9 = 3

列值小于10,因此范围在0-9之内.

我想用范围类别替换上面的数据帧值,输出应该是这样的

a  b  c
1  1  1
2  3  3
2  1  1
1  2  2
2  1  2
.......

因此,如果df中的任何值在0-3之内,则应替换为1,如果4-6内的任何值将其替换为2,依此类推.我怎样才能做到这一点？

Answer 1

pd.cut与apply ie一起使用

df.apply(lambda x : pd.cut(x,[-1,3,6,9],labels=[1,2,3]))

   a  b  c
0  1  1  1
1  2  3  3
2  2  1  1
3  1  2  2
4  2  1  2

@coldspeed建议的基于非应用的解决方案:

pd.DataFrame(pd.cut(df.values.reshape(-1,),[-1,3,6,9],labels=[1,2,3]).codes.reshape(df.shape)+1,columns=df.columns)

要么

pd.DataFrame(pd.cut(np.hstack(df.values),[-1,3,6,9],labels=[1,2,3]).codes.reshape(df.shape)+1,columns=df.columns)