Python：データフレームの情報を新しい列に移したい

Question

前提・実現したいこと

Pythonを使用しています。

• 実現したい内容 …

「　sample['way']の内容を全て1文字だけにし、sample['way_']に移したいです。apply()を4つ試したのですが、1番最後しか適用されなかったです。」

実行したコードとその結果

実行したコード

このようなコードを実行しました。

In[]:
import pandas as pd
sample = pd.DataFrame({'way':['up', 'down', 'left', 'right', 'up', 'right', 'left']})

sample['way_'] = sample['way'][sample['way']=='up'].apply(lambda x:'u' if x=='up' else 0)
sample['way_'] = sample['way'][sample['way']=='down'].apply(lambda x:'d' if x=='down' else 0)
sample['way_'] = sample['way'][sample['way']=='left'].apply(lambda x:'l' if x=='left' else 0)
sample['way_'] = sample['way'][sample['way']=='right'].apply(lambda x:'r' if x=='right' else 0)

sample

実行した結果

上記のコードを実行した結果、
以下のようになりました。

Out[]:
     way	way_
0	up	NaN
1	down	NaN
2	left	NaN
3	right	r
4	up	NaN
5	right	r
6	left	NaN

本来得たい結果

望んでいた実行結果は以下の通りです。

Out[]:
     way	way_
0	up	u
1	down	d
2	left	l
3	right	r
4	up	u
5	right	r
6	left	l

Answer 1

applyの処理は詳しくないのですが、どうも上書きされているようですね。
1行に処理をまとめてしまって、下記でいかがでしょう。

import pandas as pd
sample = pd.DataFrame({'way':['up', 'down', 'left', 'right', 'up', 'right', 'left']})
sample['way_'] = sample['way'].apply(lambda x:x[0])

sample

Answer 2

pandasで文字列にスライスを適用して任意の位置・長さの部分を抽出 | note.nkmk.me

sample['way_'] = sample['way'].str[0]

# あるいは
sample['way_'] = [i[0] for i in sample['way'].to_list()]

を用いてください。