Pandas 数据框中的新列基于现有列中的变量值

Question

我在创建一个新列时遇到了困难，该列的值基于同一数据框中现有列的值。现有列是数字，我正在尝试根据以下内容为新列提供高、中、低的分类值：

低：

med: (max-min)/3 - (max-min)/3 *2

高：> (max-min)/3 *2

仍在学习 Pandas，因此感谢您的帮助。谢谢！

编辑：

这是我尝试过的：

df_unit_day_hour['Level_Score'] = pd.cut(df_unit_day_hour['Level_Score'], q=3, labels=['low', 'medium', 'high'])

我认为这几乎是我所需要的，但我收到了一个错误 (KeyError)。会不会是因为 df_unit_day_hour['Level_Score'] 是一个浮点数？

Answer 1

听起来你想重新创建Series.cut 函数

考虑下面这个例子：

import numpy as np
import pandas as pd

df = pd.DataFrame({'val':np.random.choice(10, 10)})
df['cat'] = pd.cut(df['val'], [-1,2,5,10], labels=['low', 'medium', 'high'])
    df

   val   cat
0    6  high
1    2   low
2    7  high
3    7  high
4    8  high
5    8  high
6    9  high
7    6  high
8    2   low
9    0   low