データフレームの値から新しいカラムを作りたい

Question

前提・実現したいこと

ここに質問の内容を詳しく書いてください。
（例）PHP(CakePHP)で●●なシステムを作っています。
■■な機能を実装中に以下のエラーメッセージが発生しました。

発生している問題・エラーメッセージ

下記のデータフレームで、身長に応じてグループ分けのカラムを新たに作りたい考えております。
180以上をグループ0,170までを1,169以下を２として、df1['グループ']=●●のように、あるいはスライスを使って試行錯誤しましたが自分では解決できなかったので力を貸していだだけないでしょうか？

該当のソースコード

df1 = pd.DataFrame({
      '身長': [160, 165, 150, 180, 175, 160,160, 165, 150, 180],
      '体重': [ 70,  55,  50, 70, 60, 65,70,  55,  50, 70],
      '成績': [  3,   2,   1,  4,   5,  6, 7,   8,   9,  10],
  '過去成績': [  2,   3,   1,   5,   4,  6, 7, 8, 9, 10],
}, index = ['A', 'B', 'C', 'D', 'E'])

##追記
df1のカラム['グループ']に身長1,2番目をグループ0,グループ０を除く行の上位半分(この場合は３~6)をグループ１残りをグループ２としてデータフレームのカラムに追加する方法を教えていただきたく思います。

Answer 1

df1 = pd.DataFrame({
      '身長': [180, 175, 150],
      '体重': [ 70,  55,  50],
      '成績': [  3,   2,   1],
  '過去成績': [  2,   3,   1],
}, index = ['A', 'B', 'C'])

df1['グループ'] = df1['身長'].apply(lambda x: 0 if x >= 180 else 1 if x >= 170 else 2)
print(df1)

#
   身長  体重  成績  過去成績  グループ
A   180    70     3         2         0
B   175    55     2         3         1
C   150    50     1         1         2

Answer 2

身長からグループ値を返す関数を作ってapplyすればよいです。

import pandas as pd

def f(v):
    if v >= 180:
        return 0
    elif v >= 170:
        return 1
    return 2

df = pd.DataFrame({'tall':[180,170,160]})
df['group'] = df['tall'].apply(f)
print(df)
#   tall  group
#0   180      0
#1   170      1
#2   160      2

Comments

[tomoysh]

回答ありがとうございます。もう一つ教えていただきたいのですが、df１が30列あったとして身長上位５人をグループ0、残りの人数を半分で割った上位をグループ１、下位をグループ２とする場合はどのようにするといいのでしょうか？