python中的数据分类

Question

​

​

我有一个数据集，类似于上面使用read_csv函数读取到python的数据集。我想做的是

1. 将环列改为类列，其中1-5环为1类，5-10环为2类，依此类推。现在，我有一个如下所示的for循环 而j<4177 : if (Xj) < 9：(Xj) =1 elif (Xj)是9 x (Xj)是10：(Xj) =2：(Xj) =3 j=j+1 有什么更有效的方法来做到这一点呢？
2. 假设我想保留我的环列，而不是在类中添加另一列，那么最好的方法是什么？

Answer 1

我想你可以用cut

np.random.seed(100)
df = pd.DataFrame(np.random.randint(20, size=(20,2)), columns=['a','ring'])
print (df)
     a  ring
0    8     3
1    7    15
2   16    10
3    2     2
4    2    14
5    2    17
6   16    15
7    4    11
8   16     9
9    2    12
10   4     1
11  13    19
12   4     4
13   3     7
14  17    15
15   1    14
16   7    16
17   2     9
18  19     2
19  14    17

classes = 4

bins = np.linspace(0, classes * 5, classes + 1)
labels=['class ' + str(i + 1) for i in range(classes)]
print (bins)
[  0.   5.  10.  15.  20.]
print (labels)
['class 1', 'class 2', 'class 3', 'class 4']

df['label'] = pd.cut(df['ring'], bins=bins, labels=labels)
print (df)
     a  ring    label
0    8     3  class 1
1    7    15  class 3
2   16    10  class 2
3    2     2  class 1
4    2    14  class 3
5    2    17  class 4
6   16    15  class 3
7    4    11  class 3
8   16     9  class 2
9    2    12  class 3
10   4     1  class 1
11  13    19  class 4
12   4     4  class 1
13   3     7  class 2
14  17    15  class 3
15   1    14  class 3
16   7    16  class 4
17   2     9  class 2
18  19     2  class 1
19  14    17  class 4