pandas 複数の列からの条件付コピー

Question

python 3.5
pandas 0.22.0において

A	US	JP
0	14	22
newyork	14	22
newyork	15	22
tokyo	15	23
london	15	23

という元のDataFrameがある場合に、
Aの列の値に応じて、C1列,C2列の値を参照して新たなX列を作りたい

つまり
AがnewyorkであればUS列の値をX列に
AがtokyoであればJP列の値をX列に
したいです。

|A|US|JP|X|
|:--|:--:|--:|
|0|14|22||
|newyork|14|22|14|
|newyork|15|22|15|
|tokyo|15|23|23|
|london|15|23||

同様の質問は以下にもありますが、やや異なります。
https://teratail.com/questions/62152

Answer 1

求めたい値を返す関数を定義し、.apply(関数, axis=1)で行毎に処理すればよいです。

import pandas as pd
df = pd.DataFrame({'A':['','newyork','newyork','tokyo','london'],'US':[14,14,15,15,15],'JP':[22,22,22,23,23]})
print(df)

def getX( df):
    if df['A'] == 'newyork':
        return df['US']
    elif df['A'] == 'tokyo':
        return df['JP']
    # 該当なしはNone(NaN)でいいかな

df['X'] = df.apply( getX, axis=1)
print(df)

Comments

[jckkvs]

なるほど！　ありがとうございます。求めていた動作が実現できそうです。
月曜日に確認してBAにさせていただきます。

Answer 2

一応書いて見ましたが、これでよいのでしょうか？

import pandas as pd
import io

data = """A,C1,C2
0,14,22
1,14,22
1,15,22
2,15,23
3,15,23
"""

df = pd.read_csv(io.StringIO(data))
df['X'] = df.apply(lambda r: r.get('C{}'.format(r.A)), axis=1)
print(df)
#   A  C1  C2     X
#0  0  14  22   NaN
#1  1  14  22  14.0
#2  1  15  22  15.0
#3  2  15  23  23.0
#4  3  15  23   NaN

---

【修正分】

import pandas as pd
import io

data = """A,US,JP
0,14,22
newyork,14,22
newyork,15,22
tokyo,15,23
london,15,23
"""

df = pd.read_csv(io.StringIO(data))
conv = {'newyork': 'US', 'tokyo': 'JP', 'London':'UK'}
df['X'] = df.apply(lambda r: r.get(conv.get(r.A)), axis=1)
print(df)
#         A  US  JP     X
#0        0  14  22   NaN
#1  newyork  14  22  14.0
#2  newyork  15  22  15.0
#3    tokyo  15  23  23.0
#4   london  15  23   NaN

Comments

[jckkvs]

ご回答ありがとうございます。
途中で内容を変えて申し訳ないのですが、簡略化して表記していましたが
実際には1・２がそのままC1列、C2列に対応するものではないので変更させていただきました。

この場合、
r: r.get('C{}'.format(r.A)), axis=1
をどのように記述すればよろしいでしょうか。

[magichan]

ん？質問変わってる？

[magichan]

修正を書きますので、ちょっとお待ちを

[magichan]

書きました。
変換用のDictを準備するとよいかと思います。

[jckkvs]

ご回答ありがとうございました。　無事動作いたしました