pythonのdataframeにおけるif条件抽出

Question

▼実現したいこと
特定の列の中身が任意の値だった場合に、
別の列の数値を引用して当てはめるif条件式を作りたいと考えております。

▼サンプルデータフレームの作成
以下のようなデータフレームを作ったうえで、

import pandas as pd

list = [
[40, 'A'], 
[10, 'B'], 
[15, 'C'], 
[20, 'A'], 
[80, 'B'], 
[35, 'C']]
df = pd.DataFrame(list)
df.columns = ['lap','area']

df['lap_A'] = 0
df['lap_B'] = 0
df['lap_C'] = 0

カラム'area'の中身が'A'だった場合は、'lap_A'の中に、'lap'の数値を当てはめる、
カラム'area'の中身が'B'だった場合は、'lap_B'の中に、'lap'の数値を当てはめる、
カラム'area'の中身が'C'だった場合は、'lap_C'の中に、'lap'の数値を当てはめるためのif条件式を作りました。

df["area"] == 'A'] = 'lap'
df["area"] == 'B'] = 'lap'
df["area"] == 'C'] = 'lap'

▼実行結果
エラー内容が以下の通りとなります。

SyntaxError: invalid syntax

こちらのif条件式の修正方法をご教示いただけませんでしょうか。

Answer 1

df["area"] == 'A'] = 'lap'
df["area"] == 'B'] = 'lap'
df["area"] == 'C'] = 'lap'

SyntaxError: invalid syntaxの原因は[が足りないからだと思います。

Comments

[color_8]

ご指摘有難うございました！解決いたしました。

Answer 2

df.loc[df['area'] == 'A', 'lap_A'] = df['lap']
df.loc[df['area'] == 'B', 'lap_B'] = df['lap']
df.loc[df['area'] == 'C', 'lap_C'] = df['lap']

Comments

[color_8]

コードの指定有難うございました。
おかげさまで解決いたしました！