【NumPy】配列の分割(縦方向・横方向)

この記事では、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]])]

おすすめ記事

Python入門 サンプル集
NumPy入門 サンプル集

関連記事