如何根据另一个值编辑列值？

Question

我的数据名是这样的：

val    type
10     new
70     new
61     new
45     old
32     new
77     mid
11     mid

对于“类型”列中的值，如果值是新的，我想根据 val 列对其进行编辑。如果 val 中的 value = 20 且 = 50，则必须为 new-3。所以想要的结果是：

val    type
10     new-1
70     new-3
61     new-3
45     old
32     new-2
77     mid
11     mid

怎么做？

Answer 1

我会使用pd.cut:

>>> df['type'] = df['type'].where(df['type'].ne('new'), df['type'] + '-' + pd.cut(df['val'], [0, 20, 50, float('inf')], labels=[1, 2, 3]).astype(str))
>>> df
   val   type
0   10  new-1
1   70  new-3
2   61  new-3
3   45    old
4   32  new-2
5   77    mid
6   11    mid
>>>