複数条件で抽出したindex（dataframe）をグループ分けしたい

dfデータ

class,sex,weight,height,time
A,F,45,150,85
A,M,50,160,80
A,F,55,155,74
B,M,78,180,90
B,F,51,158,65
B,M,40,155,68
C,F,80,185,90
C,M,86,175,81
C,F,52,162,73

python
1groupA = df[df["time"]>=85][df["weight"]>=75]
2groupB = df[df["time"]<=70][df["weight"]<=60]
3

抽出したデータグループ分けし
新たに追加したgroupカラムにgroupA=1,groupB=2,その他を０とするにはどうしたらよいでしょうか？

試したこと

python
1df["group"]=0
2df[df["time"]>=85][df["weight"]>=75]["group"]=1
3df[df["time"]>=85][df["weight"]>=75]["group"]=2
4

df["group"]の中身にうまく代入されません
0
0
0
0
0
0
0
0
0

最終的に
df["group"]が
0
0
0
1
2
2
1
0
0

になるようにしたいです。

行動規範の内容に同意します

回答1件

ベストアンサー

Python
1df['group'] = 0
2df.loc[(df['time'] >= 85) & (df['weight'] >= 75), 'group'] = 1
3df.loc[(df['time'] <= 70) & (df['weight'] <= 60), 'group'] = 2
4
5"""
6  class sex  weight  height  time  group
70     A   F      45     150    85      0
81     A   M      50     160    80      0
92     A   F      55     155    74      0
103     B   M      78     180    90      1
114     B   F      51     158    65      2
125     B   M      40     155    68      2
136     C   F      80     185    90      1
147     C   M      86     175    81      0
158     C   F      52     162    73      0
16"""