SAS中的proc格式等效于python

Question

我想要

proc format; 
 value RNG

low - 24  = '1'
24< - 35  = '2'
35< - 44  = '3'
44< - high ='4'

我在python大熊猫中需要它。

Answer 1

如果您正在寻找等效的映射功能，则可以使用类似的内容。

df = pd.DataFrame(np.random.randint(100,size=5), columns=['score'])

print(df)

输出：

   score
0     73
1     90
2     83
3     40
4     76

现在，我们可以为数据框中的score列应用合并功能，并在同一数据框中创建新列。

def format_fn(x): 
    if x < 24:
        return '1'
    elif x <35:
        return '2'
    elif x< 44:
        return '3'
    else:
        return '4'

df['binned_score']=df['score'].apply(format_fn)

print(df)

输出：

   score binned_score
0     73            4
1     90            4
2     83            4
3     40            3
4     76            4