【Python】NumPy配列の覚え書き
NumPy配列のサイズの確認方法
行列のサイズを取得したいときには,numpy.shapeを用いる。
import numpy as np import pandas as pd nparr = np.full((3, 2), 0) # (行,列)の順で指定 print(nparr) # [[0 0 0] # [0 0 0] # [0 0 0]] #行列サイズをまとめて取得 column, row = nparr.shape #行もしくは列毎に取得 column2 = nparr.shape[0] row2 = nparr.shape[1]
NumPy配列の結合方法
- 次元数が増加しないような結合の場合
numpy.concatenate()を用いて結合する。
第一引数には,結合したい配列をリスト,タプルとして指定
第二引数には,結合方向の軸を指定(0 = 行,1=列,2=高さ,...的な)
○ベクトル同士の結合
vector1 = np.full(3, 0) # [0 0 0] vector2 = np.full(3, 1) # [1 1 1] vector0 = np.concatenate([vector1,vector2]) #np.concatenate([vector1,vector2],0)でも同じ print(vector0) #[0 0 0 1 1 1]
○行列同士の結合
arr1 = np.full((3,3), 0) #[[0 0 0] # [0 0 0] # [0 0 0]] arr2 = np.full((3,3), 1) #[[1 1 1] # [1 1 1] # [1 1 1]] arr0_0 = np.concatenate([arr1,arr2],0) print(arr0_0) #[[0 0 0] # [0 0 0] # [0 0 0] # [1 1 1] # [1 1 1] # [1 1 1]] arr0_1 = np.concatenate([arr1,arr2],1) print(arr0_1) #[[0 0 0 1 1 1] # [0 0 0 1 1 1] # [0 0 0 1 1 1]]