2011-11-18

*1321604904*Numpyのndarrayは空でなくても偽になる

Pythonのリストの場合、ifの条件式に入れてboolに変換された場合は日本語で言うと「空かどうか」になる。

if []: print not empty

if 0: print not empty

not empty

print bool([]), bool(0)

False True

でもNumpyのndarrayはそうではない。

print bool(array([])), bool(array(0))

False False

これは、ndarrayは「1要素で中身が0の時にはゼロである」と設計されているせい。

array(0).__nonzero__()

False

Pythonのリストの挙動とは整合しないけど、まあゼロベクトルがゼロと判定されるのはアリかなぁ、などと思ったが…

array(0, 0, 0).__nonzero__()

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

えっ。

*1321611727*NumPyで特定の条件を満たす行、列を除去する方法

NumPyではスライシングで特定の条件を満たす要素だけ選択できる。逆の条件で使えば特定の要素だけ除去するのも簡単。

|python|

array(range(5))

array(0, 1, 2, 3, 4)

v = _

vv != 2

array(0, 1, 3, 4)

これは「v != 2」の部分で演算子がオーバーロードされていて、真偽値のarrayになっている。だからこれは他のarrayに対しても使える。

|python|

v != 2

array( True, True, False, True, True, dtype=bool)

randn(5)

array( 1.96182557, 1.31524788, -0.09140969, -0.80362045, 0.02612784)

_v != 2

array( 1.96182557, 1.31524788, -0.80362045, 0.02612784)

ベクトルが縦になっているなら1つ目の軸に「:」(全部)を指定して2番目にスライスを指定すればよい。

|python|

randn(5).reshape(1, 5)

array(-0.86985953, 0.59620319, -1.25975869, 1.87290875, -0.44375606)

_:, v != 2

array(-0.86985953, 0.59620319, 1.87290875, -0.44375606)

ここまではa href='http://www.scipy.org/Tentative_NumPy_Tutorial#head-864862d3f2bb4c32f04260fac61eb4ef34788c4c'Tentative NumPy Tutorial/aにも書いてあるけど、今回やりたいことは正方行列の行と列と両方でこの除去をすること。

|python|

array(range(25)).reshape(5, 5)

array([ 0, 1, 2, 3, 4,

mat = _

単純に行と列の両方にこのスライシングを指定したらいいんじゃないか？って思うけどこれは間違い。対角成分だけになってしまう。

|python|

matv != 2, v != 2

array( 0, 6, 18, 24)

片方をスライスしてからもう片方をスライスすればOK。

|python|

array([ 0, 1, 3, 4,