Python:データフレームの各要素に対してある計算をして、列を追加し、計算結果を格納する
解決済
回答 1
投稿
- 評価
- クリップ 0
- VIEW 2,608
前提・実現したいこと
Pandasの 各要素の演算方法ついて質問です。
例えばこんなデータフレームがあるとします。
//Test
hoge factors
sample1 1 1
sample2 1 1
sample3 2 3
//Dictionary
factor1 factor2
cat 1 2
dog 3 4
データフレーム:Testの列:factorsの値をデータフレーム:Dictionaryの要素と照らし合わせて
catかdogかを判定した後に, このデータフレームの末尾にresultという
列名をつけて追加したいと思います。
アウトプットイメージとしては次のとおりです。
hoge factors result
sample1 1 1 cat
sample2 1 1 cat
sample3 2 3 dog
「試したこと」の欄のようにapplyを使って書いてみましたが、
うまく行きません。
上手いを教えていただければと思います。
発生している問題・エラーメッセージ
なし
試したこと
applyを使って下のように書いてみましたが,
色々調べてみると applyはSeriesごとの計算しか出来ないみたいです。
(間違ってたらすみません)
import pandas as pd
# 辞書として使うデータフレームを定義する
df_dict= pd.DataFrame( data=[[1,2],[3,4]], index = ['cat','dog'],columns=['factor1','factor2']).stack()
# 判定用のデータを用意する
df_test = pd.DataFrame(data=[[1,1],[1,1],[2,3]], index = ['sample1','sample2','sample3'],columns=['hoge','factors'])
# 判定するための関数を定義
def ident(valu):
spec = list(df_dict[df_dict == valu].index)
return spec[0][0]
# 判定する
df_test['result'] = df_test.apply(### ここの書き方が分からないので思ったことが出来ない? ###)
-
気になる質問をクリップする
クリップした質問は、後からいつでもマイページで確認できます。
またクリップした質問に回答があった際、通知やメールを受け取ることができます。
クリップを取り消します
-
良い質問の評価を上げる
以下のような質問は評価を上げましょう
- 質問内容が明確
- 自分も答えを知りたい
- 質問者以外のユーザにも役立つ
評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。
質問の評価を上げたことを取り消します
-
評価を下げられる数の上限に達しました
評価を下げることができません
- 1日5回まで評価を下げられます
- 1日に1ユーザに対して2回まで評価を下げられます
質問の評価を下げる
teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。
- プログラミングに関係のない質問
- やってほしいことだけを記載した丸投げの質問
- 問題・課題が含まれていない質問
- 意図的に内容が抹消された質問
- 過去に投稿した質問と同じ内容の質問
- 広告と受け取られるような投稿
評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。
質問の評価を下げたことを取り消します
この機能は開放されていません
評価を下げる条件を満たしてません
質問の評価を下げる機能の利用条件
この機能を利用するためには、以下の事項を行う必要があります。
- 質問回答など一定の行動
-
メールアドレスの認証
メールアドレスの認証
-
質問評価に関するヘルプページの閲覧
質問評価に関するヘルプページの閲覧
checkベストアンサー
0
キーとする列がどれになるのかわからなかったのですが、df_testのfactorsの要素とdf_dictのfactor1の要素がマッチしていればよいのでしょうか?
上記を前提としているのであれば、こんな書き方でいかがでしょうか。
import pandas as pd
df_dict = pd.DataFrame(data=[[1, 2], [3, 4]],
index=['cat', 'dog'],
columns=['factor1', 'factor2'])
df_test = pd.DataFrame(data=[[1, 1], [1, 1], [2, 3]],
index=['sample1', 'sample2', 'sample3'],
columns=['hoge', 'factors'])
df_test['result'] = df_test['factors'].map(
lambda x: df_dict[df_dict['factor1'] == x].index[0])
投稿
-
回答の評価を上げる
以下のような回答は評価を上げましょう
- 正しい回答
- わかりやすい回答
- ためになる回答
評価が高い回答ほどページの上位に表示されます。
-
回答の評価を下げる
下記のような回答は推奨されていません。
- 間違っている回答
- 質問の回答になっていない投稿
- スパムや攻撃的な表現を用いた投稿
評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。
15分調べてもわからないことは、teratailで質問しよう!
- ただいまの回答率 88.09%
- 質問をまとめることで、思考を整理して素早く解決
- テンプレート機能で、簡単に質問をまとめられる
2017/01/28 06:03
説明が不足していましたが,キー列はどちらでも大丈夫です.
「lambda x: df_dict[df_dict == x].index[0]」という風にして,手元で試してみました.
ありがとうございました!