Python 3.8.5でpandasを使って、データの集計を行いたい

Question

前提・実現したいこと

Python 3.8.5でpandasを使って、データの集計を行う
複数行を、特定の列(state)でgroupして、pointを集計したいが、
特定の列(state)に特定文字(JAPAN)を含むレコードは、同一のレコードとみなしてgroup化したい

該当のソースコード

import pandas as pd
df = pd.read_csv('C:\0128\raw\raw.csv')
print(df[df['state'].str.contains('JAPAN')])

このように集計したい

name,age,state,point
Alice,24,JAPAN,100
Bob,42,CANADA,92
Charlie,18,AMERICA,70
Dave,68,JAPANESE,100
Ellen,24,JAPANESE,100
Frank,30,FRANCE,57

↓↓

state,point
JAPAN,300
CANADA,92
AMERICA,70
FRANCE,57

Comments

[takutakuya]

JAPANとJAPANESEは同一とみなすんですか？

[usaginote1020]

はい。JAPANという文字を含むものは同一とみなしたいです。

Answer 1

import pandas as pd
import io

txt = """
name,age,state,point
Alice,24,JAPAN,100
Bob,42,CANADA,92
Charlie,18,AMERICA,70
Dave,68,JAPANESE,100
Ellen,24,JAPANESE,100
Frank,30,FRANCE,57
"""

df = pd.read_csv(io.StringIO(txt))
# df = pd.read_csv('C:\0128\raw\raw.csv')
df.loc[df['state'].str.contains("JAPAN"), 'state'] = "JAPAN"
dfg = pd.DataFrame(df.groupby('state')['point'].sum())
print(dfg)

         point
state
AMERICA     70
CANADA      92
FRANCE      57
JAPAN      300

Comments

[usaginote1020]

ありがとうございました。希望通りに出来ました