pandas 関数の使い方について教えて下さい。

Question

python初心者です。連投の質問で恐縮です。

defで関数を定義して、データフレームに値を入れられるようにしたのですが、
1～6の数字は繰り返しなので、ここをもっと短く記載をするようするにはどう
書いたら良いでしょうか。

def kansu(X):
if X == 1:
return 1
elif X ==2:
return 2
elif X ==3:
return 3
elif X ==4:
return 4
elif X==5:
return 5
elif X==6:
return 6
elif X >=7:
return 7

df['月']=df['月1'].apply(kansu)

Comments

[meg_]

コードは「コードの挿入」で記入しましょう。

[gomagoma1122]

ご指摘ありがとうございます。良くわかっておらず大変申し訳ありません。
次回からは気を付けさせて頂きます。

[meg_]

> 次回からは気を付けさせて頂きます。
質問は編集できます。

Answer 1

def kansu(X):
    return min(X, 7)

です。

Comments

[gomagoma1122]

ご回答頂きありがとうございます。
このコードで無事に出来ました。min関数を使うのですね。
大変勉強になりました。

Answer 2

df['月']=df['月1'].apply(kansu)

pandas.core.frame.where(cond, other=nan, inplace=False, ...

Replace values where the condition is False.

import pandas as pd

df = pd.DataFrame({
  '月1': range(1, 12)
})

df['月'] = df['月1'].where(df['月1'].lt(7), 7)

pd.set_option('display.unicode.east_asian_width', True)
print(df)

#
    月1  月
0     1   1
1     2   2
2     3   3
3     4   4
4     5   5
5     6   6
6     7   7
7     8   7
8     9   7
9    10   7
10   11   7

Comments

[gomagoma1122]

whereを使ったことが無かったのですが、この記述で無事に解決しました。
これから勉強してみます。大変勉強になりました。ありがとうございます。