データフレームで列を追加（条件に応じて値を変える）

Question

jupyter notebook で列を追加したいです。
追加した列の値は条件に応じて変更したいです。

下記のデータフレームがあるとして

部署
0 A
1 B
2 A
3 A

下記のように単価を追加して、
部署Aなら4000円、部署Bなら5000円のようにしたいです。

部署 単価
0 A　　　4000
1 B　　　5000
2 A　　　4000
3 A　　　4000

よろしくお願い致します。

Answer 1

部署に応じた価格を定義した dictやSeriesのデータを準備してpandas.Series.map()を使うと良いかと思います。

https://pandas.pydata.org/pandas-docs/stable/generated/pandas.Series.map.html

import pandas as pd

df = pd.DataFrame({'部署':['A','B','A','B']})

price = {'A':4000, 'B':5000}

df['単価'] = df['部署'].map(price)

print(df)
#  部署    単価
#0  A  4000
#1  B  5000
#2  A  4000
#3  B  5000

Comments

[satoshi07]

ありがとうございます。
解決しました。