2018年5月31日木曜日

pythonで行列の行、列のインデックス、値を全てイテレートする方法

# pythonで行列の行、列のインデックス、値を全てイテレートする方法
# numpyのndenemurateを使う
import numpy as np
X = np.array([
    ['a0', 'a1', 'a2'],
    ['b0', 'b1', 'b2']
])
# イテレートする index はタプルになる
for index, value in np.ndenumerate(X):
    print(index, value)
# このようにもできる
# i, jを分けて イテレート
for (i,j), value in np.ndenumerate(X):
    print((i,j),value)
# i行j列:としたければ インデックスを+1する
for (i,j), value in np.ndenumerate(X):
    print('{}行{}列は{}'.format(i+1,j+1,value))
# 1行1列はa0
# 1行2列はa1
# 1行3列はa2
# 2行1列はb0
# 2行2列はb1
# 2行3列はb2

0 件のコメント:

コメントを投稿