Pythonで特定の値に達した時、それ以降の行を抽出し,さらにそれを横並びにしたいです．

Question

以下に以前教えていただいたプログラムを示します．
説明はこのURLの通りです．(https://teratail.com/questions/197955)

今回はこの条件を3つに，抜き出す行を4つにしたいです．
以下のように変更すればできると思ったのですができませんでした，
どのようにすれば理想のようにいくのでしょうか．

#変更したもの
import pandas as pd
import io


data = """
1,100,0.145
2,110,0.146
3,120,0.147
4,130,0.148
5,140,0.149
6,150,0.150
7,145,0.150
8,140,0.149
9,130,0.150
10,120,0.151
11,150,0.160
12,140,0.159
13,130,0.160
14,120,0.161
"""

df = pd.read_csv(io.StringIO(data), header=None)


target_dfs = []
count = 0
for idx,row in df.iterrows():
    if (count == 0) and (row[2] != 0.146) and (row[2] != 0.150) and (row[2] != 0.160):
        continue
    elif count == 0:
        p = df.index.get_loc(idx)
        target_dfs.append(df.iloc[p:p+4].reset_index(drop=True))
    count = (count + 1) if (count < 4) else 0

target = pd.concat(target_dfs, axis=1)

print(target)

#結果
   0    1      2   0    1      2     0      1      2
0  2  110  0.146   7  145  0.150  13.0  130.0  0.160
1  3  120  0.147   8  140  0.149  14.0  120.0  0.161
2  4  130  0.148   9  130  0.150   NaN    NaN    NaN
3  5  140  0.149  10  120  0.151   NaN    NaN    NaN

#理想の結果
   0    1      2   0    1      2     0      1      2
0  2  110  0.146   6  150  0.150    11    150  0.160
1  3  120  0.147   7  145  0.150    12    140  0.159
2  4  130  0.148   8  140  0.149    13    130  0.160
3  5  140  0.149   9  130  0.150    14    120  0.161

Comments

[t_obara]

仕様を整理して再度ご提示ください。

[godagoda]

大変失礼致しました
このプログラムでは、質問最初にあるデータの3列目が最初に0.146、0.150、0.160となったとき、それぞれ、それ以降の行を4つだけ抽出してそれぞれを新しい列として出力したいです。

Answer 1

どうもカウントの条件に間違いがあるようですね・・

    count = (count + 1) if (count < 4) else 0

を

    count = (count + 1) if (count < 3) else 0

に修正してみてください。

このバグを埋め込んだのはもしかしたら私でしょうかね・・・・・・・

であれば大変申し訳ありません

Comments

[godagoda]

いえ、そこは今回は4行抜き出そうと、私が変えたものです。
説明足らずで申し訳ありません。