【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

def main():
    # 配列の宣言・初期化
    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]])]

if __name__ == '__main__':
    main()
    

おすすめ記事

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

シェア&フォローお願いします!