Pythonで配列の1行にあるn個の要素から大きい順に3個の値を抽出する

Question

###Pythonで配列の1行にあるn個の要素から大きい順に3個の値を抽出する方法

初学者で、以下のようなデータで範囲を指定し、
処理前から処理後を抽出する方法はないかと思い、質問させていただきます。

(処理前)

|S1(文字列)|a|b|c|d|e|f|g|h|S2(文字列)|
|:--:|--:|
|○○○○|1|2|3|4|5|6|7|8|○○○○|
|○○○○|2|4|6|8|10|12|14|16|○○○○|

(処理後)
|S1(文字列)|上位1|上位2|上位3|S2(文字列)|
|:--:|--:|
|○○○○|8|7|6|○○○○|
|○○○○|16|14|12|○○○○|

ahまでは同種の要素で、処理後はカラム名を別に指定したいと考えております。
ahまでは、同じ値にならない場合と考えております。

拙い質問で申し訳ありませんが、ご教示いただければと思います。
よろしくお願い致します。

###試したこと
最大値を得られますので、
これとfor文を使って、抽出する方法があるかとは思うのですが・・・

###補足情報(言語/FW/ツール等のバージョンなど)
Pythonは3.5.4を使用しています。
Pandasは0.21.0

Answer 1

書いてみました。こんな感じでどうでしょうか。

もう少しシンプルになって欲しいところ・・・

import pandas as pd

df = pd.DataFrame([['S1_1',1,2,3,4,5,6,7,8,'S2_1'],
                  ['S1_2',2,4,6,8,10,12,14,16,'S2_2']],
                  columns=['S1','a','b','c','d','e','f','g','h','S2'])
print(df)
#      S1  a  b  c  d   e   f   g   h    S2
# 0  S1_1  1  2  3  4   5   6   7   8  S2_1
# 1  S1_2  2  4  6  8  10  12  14  16  S2_2

rank = df.iloc[:,1:9].apply(lambda s:s.sort_values(ascending=False)[:3].reset_index(drop=True), axis=1)
rank.columns = ['R1', 'R2', 'R3']
res = pd.concat([df['S1'], rank, df['S2']], axis=1)

print(res)
#      S1  R1  R2  R2    S2
# 0  S1_1   8   7   6  S2_1
# 1  S1_2  16  14  12  S2_2

Comments

[mkgrei]

細かいかもしれませんが、少しインデックスがずれています。
rank = df.iloc[:,1:8].apply(lambda s:s.sort_values(ascending=False)[:3].res
↓
rank = df.iloc[:,1:9].apply(lambda s:s.sort_values(ascending=False)[:3].res

[magichan]

おっホントだ
結果も間違ってる・・・・
指摘、有り難うございました。直しておきます。

[wiz_noise]

magicchanさんご回答ありがとうございます。質問に沿った回答で大変助かりました。

Answer 2

こんな感じではどうでしょう。

import numpy as np
import pandas as pd

r = np.random.randint(10, size=(10, 5))
df = pd.DataFrame(r)

N = 3
c = df.loc[0, :].nlargest(N)

print('DataFrame')
print(df)
print('FirstRow')
print(df.loc[0, :])
print('LargestNInFirstRow')
print(c)
print('NewDataFrame')
print(df.loc[:, c.index])