#前提
いつもお世話になっております。またデータフレームの操作について質問させて頂きます。
次のようなデータフレームがあるとします。
データフレームの列名はそれぞれ
- cost : ある材料を入れて機械を動かしたときのコスト
- machine : 機械の種類
- material_ID : 材料の種類
を表しているとします。
#実現したいこと
箇条書すると
0. 同じ'machine'(機械の種類)で括り,
0. その中で'cost'が一番小さいものだけを残し
0. それ以外はすべて削除する
ということがしたいです。
最終型としては次のようになるはずです。
cost machine material_ID 0 100 A 1 4 100 B 1 6 200 C 2
#試したこと
最初はグループ機能(DataFrame.groupby)などを使って実現できないかを模索しましたが,うまく行かず結局Forループを駆使してみましたが,それでもうまく行きませんでした。
下にForループを使って試したコードを記載しますが,別にForループにこだわっている訳ではなく、なるべく簡単に書ければいいなと思っています。個人的にはグループを使うのが一番早いかな、と思っておりますが、妙案があれば教えていただければと思っています。
python
1import pandas as pd 2 3# データフレームの定義 4df = pd.DataFrame( 5 {'machine': ['A', 'A', 'B', 'B', 'B', 'C', 'C'], 6 'cost': [100, 110, 120, 150, 100, 210, 200], 7 'material_ID': [1, 2, 2, 3, 1, 1, 2]}) 8 9# Forループでひたすら削る 10for i in range(len(df)): 11 if i > 0 and df.ix[i,'machine'] == df.ix[i-1,'machine']: 12 if df.ix[i,'cost'] > df.ix[i-1,'cost'] : 13 df = df.drop(i) 14 else: 15 df = df.drop(i-1) 16 17print(df)
エラーメッセージは次のとおりです。
KeyError: 1
回答3件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2017/02/04 22:33