np.arrayを特定の要素で分割したい

Question

実現したいこと

このような配列があるとします。

x = [[1,  2,  3]
     [4,  5,  6]
     [7,  8,  9]
     [10, 11, 12]]

これを

section = [7, 11]

を使って, x[:, 2]の列(3, 6, 9, 12)を基準に配列全体を分割したいです。

どういうことかというと、

[[1,  2,  3]
 [4,  5,  6]]
              <- 7で分割
[[7,  8,  9]]
              <- 11で分割
[[10, 11, 12]]

このような配列を最終的に得たいです。

どうすれば可能でしょうか？

Comments

[tiitoi]

section = [10, 11] が配列が 2行/1行/1行 で分割されたことと、どのように関係するのでしょうか？

[takuntakun]

すみません。指摘ありがとうございます。
sectionを間違いました。
section = [7, 11]のような場合を想定しています。

Answer 1

import numpy as np

x = np.array(
    [[1,  2,  3],
     [4,  5,  6],
     [7,  8,  9],
     [10, 11, 12]])

section = [11, 7, 12, 13]

# 値の存在する行位置を集合として取得
rows = {0, len(x)} # 先頭と末尾は必ず含める
for val in section:
    idx = np.where(x==val)
    if len(idx[0]):
        rows.add(idx[0][0])

rows = sorted(rows)# 昇順リストに並べ替え

for s,e in zip(rows,rows[1:]):
    print(x[s:e])
#[[1 2 3]
# [4 5 6]]
#[[7 8 9]]
#[[10 11 12]]

Answer 2

x = np.array([[1,  2,  3],
              [4,  5,  6],
              [7,  8,  9],
              [10, 11, 12]])
# print(x)
# print(x.shape)
section = [7, 11]

def sectionBy(x, section):
    groupsRowIdx = [[0,1]]
    for i in range(x.shape[0]):
        for s in section:
            r = groupsRowIdx[-1]
            if s in x[i]:
                r[1] = i
                groupsRowIdx.append([i,i+1])
            # print(groupsRowIdx)
    groups = [x[l:r] for [l, r] in groupsRowIdx]
    # print(groups)
    return groups

sectionBy(x, section)