从其他列派生 DataFrame 列而不编写任何循环

Question

我有一个包含两列（动词和出现）的 DataFrame。我能够创建一个新列来确定动词的字符数（即长度）：

df['length'] = df['verb'].str.len()

第二个要求是创建一个包含文本的新列。如果ocurrence等于 1，则写'Unique'; 如果ocurrence小于或等于5，则写'Medium'；否则'High'……

...这是我迄今为止编写的代码...

df['class'] = 'Unique' if df['ocurrence'] == 1 else 'Medium' if df['ocurrence'] <= 5 else 'High'

......但它不起作用。

Answer 1

使用pd.cut：df['class'] = pd.cut(df.occurrence, bins=[0,1,5,np.inf], labels=['Unique','Medium','High'])例如：df = pd.DataFrame({'occurrence':np.random.randint(0,10,10)})>>> df   occurrence0           51           12           63           74           55           76           77           18           29           7df['class'] = pd.cut(df.occurrence, bins=[0,1,5,np.inf], labels=['Unique','Medium','High'])>>> df   occurrence   class0           5  Medium1           1  Unique2           6    High3           7    High4           5  Medium5           7    High6           7    High7           1  Unique8           2  Medium9           7    High