【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]]

参考元リンク

note.nkmk.me