データフレームの出力の並びを変えたいです．

Question

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

このとき，最後の結果が
5
6
7
10
11
12
と，いう順番に出力されていますが，これを，
5 10
6 11
7 12
と，いう順番で出力したいです．
どのようにプログラムを変えたらよいでしょうか．

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 = ((df.iloc[:, 2] == 0.15) | (df.iloc[:, 2] == 0.16))
result = df[target | target.shift(1) | target.shift(2)]

target_idx = []
count = 0
for idx,row in df.iterrows():
    # 範囲外で、条件に合致しない場合はなにもしない
    if (count == 0) and (row[2] != 0.15) and (row[2] != 0.16):
        continue
    # 範囲内の場合はリストに追加
    elif count < 3:
        target_idx.append(idx)
    count = (count + 1) if (count < 3) else 0

target = df.loc[target_idx]
#     0    1      2
#5    6  150  0.150
#6    7  145  0.150
#7    8  140  0.149
#10  11  150  0.160
#11  12  140  0.159
#12  13  130  0.160

Answer 1

現状のコードを改造するとすると、こんな感じでしょうか。

変更点はこれまでは

• 「対象となる行を target_idx に追加して最後にDataFrameから抜き出す」

としていた部分を

• 「対象となるブロック(３行)を datget_dfs に追加して、最後に横結合する」

と変更しております。

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.15) and (row[2] != 0.16):
        continue
    # 範囲内の場合はリストに追加
    elif count == 0:
        p = df.index.get_loc(idx)
        target_dfs.append(df.iloc[p:p+3].reset_index(drop=True))
    count = (count + 1) if (count < 3) else 0

target = pd.concat(target_dfs, axis=1)
print(target)
#   0    1      2   0    1      2
#0  6  150  0.150  11  150  0.160
#1  7  145  0.150  12  140  0.159
#2  8  140  0.149  13  130  0.160

Comments

[godagoda]

ありがとうございます．成功しました．

[godagoda]

もう少し発展させたい思っております．新しく質問しましたので回答していただけると幸いです．