在熊猫中分配数字范围-Java 学习之路

我正在研究数字范围的分布，例如0-1,2-3,4-6和> = 7我有以下数据框

df = pd.DataFrame()
df['T1'] =[0,2,0,3,4,5,1]
df ['T2']= [1,2,3,0,2,3,3]
df['TT'] = df.T1+df.T2

我想创建一个新列来确定范围，我写了如下：

U0_1 = df ['TT']<=1
U2_3 = df ['TT']>1 & df ['TT']<=3
U4_6 = df ['TT']>3 & df ['TT']<=6
df ['TG'] = np.select([U0_1,U2_3,U4_6],['TG_0-1','TG_2-3','TG_4-6'],default = 'TG_7>=')

但它向我显示以下错误：

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

我可以向大家寻求建议吗？

谢谢，

泽普 .

1 回答

您可以使用 pd.cut ，我认为这是此类问题的更好方法

pd.cut(df.TT,[0,1,3,6,np.inf],labels=['TG_0-1','TG_2-3','TG_4-6','TG_7>='])
0    TG_0-1
1    TG_4-6
2    TG_2-3
3    TG_2-3
4    TG_4-6
5    TG_7>=
6    TG_4-6
Name: TT, dtype: category

要修复代码，请添加 ()

U0_1 = df ['TT']<=1
U2_3 = (df ['TT']>1) & (df ['TT']<=3)
U4_6 = (df ['TT']>3) & (df ['TT']<=6)
np.select([U0_1,U2_3,U4_6],['TG_0-1','TG_2-3','TG_4-6'],default = 'TG_7>=')
array(['TG_0-1', 'TG_4-6', 'TG_2-3', 'TG_2-3', 'TG_4-6', 'TG_7>=',
       'TG_4-6'], dtype='<U6')

回复于 2024-05-05T18:43:42+08:00

在熊猫中分配数字范围

1 回答

相关问题