使用assign方法创建列

Question

假设我有以下数据帧：

我想创建一个基于列a的新列，并根据列a的值创建组。

df = pd.DataFrame([['1', 3],['2', 5],['3', 20],['4',8],['5',7],['6',13],['7',33],['8',2]], columns=['index', 'A'])
df['groups'] =  df['A'].apply(lambda x: 'high' if x>15 else 'medium' if 15>=x>10 else 'low')

我怎么能使用assign做同样的事情呢？

df = df\
.assign(groups = ?)

Answer 1

使用assign()方法：

df=df.assign(groups =df['A'].apply(lambda x: 'high' if x>15 else 'medium' if 15>=x>10 else 'low'))

或

value=df['A'].apply(lambda x: 'high' if x>15 else 'medium' if 15>=x>10 else 'low')
df=df.assign(groups=value)

现在，如果您打印df，您将得到所需的输出：

#output

    index   A   groups
0   1       3   low
1   2       5   low
2   3       20  high
3   4       8   low
4   5       7   low
5   6       13  medium
6   7       33  high
7   8       2   low