この記事では、Python言語とNumPyを用いて、配列を分割する方法をソースコード付きで解説します。
配列の分割（縦方向・横方向）
NumPy配列では配列の要素を配列を分割する機能が用意されています。
書式
| – | 説明 |
|---|---|
| y = numpy.vsplit(x, n) | 配列xを縦方向にn個に分割します。 |
| y = numpy.hsplit(x, n) | 配列xを横方向にn個に分割します。 |
ソースコード
サンプルプログラムのソースコードです。
# -*- coding: utf-8 -*-
import numpy as np
# 配列の宣言・初期化
x = np.array([[1, 2, 3],
[4, 5, 6]])
# 縦方向に分割
print(np.vsplit(x,2)) # [array([[1, 2, 3]]), array([[4, 5, 6]])]
# 横方向に分割
print(np.hsplit(x,3)) # [array([[1], [4]]),
# array([[2],[5]]),
# array([[3], [6]])]
コメント