スカラーをヌンプイ配列と比較する

Question

私がやっていることは、Pythonの断片的な関数に基づいてテーブルを作ることです。例えば、私はこのコードを書いたと言う：

import numpy as np 
from astropy.table import Table, Column 
from astropy.io import ascii 
x = np.array([1, 2, 3, 4, 5]) 
y = x * 2 
data = Table([x, y], names = ['x', 'y']) 
ascii.write(data, "xytable.dat") 
xytable = ascii.read("xytable.dat") 
print xytable 

これは予想通り、それはx値1〜5及びy値は2、4、6、8、10

を有するテーブルを印刷動作しかしが3以下の場合にのみyをx * 2とし、それ以外の場合はyをx + 2とする場合はどうすればよいですか？

私が追加した場合：

if x > 3: 
    y = x + 2 

それは言う：複数の要素を持つ配列の

真理値があいまいです。 a.any（）またはa.all（）を使用する

どのようにしてパーツ単位で機能するようにテーブルをコーディングするのですか？スカラーとナンパー配列を比較するにはどうすればいいですか？

Answer 1

あなたはおそらくnumpy.where()を使用することができます。

In [196]: y = np.where(x > 3, x + 2, y) 

In [197]: y 
Out[197]: array([2, 4, 6, 6, 7]) 

コードは、上記の取得仕事は完全にベクトル化された方法。このアプローチは、リスト内包表記と型変換を使用するよりも、一般的に効率的（そしておそらくエレガント）です。

Answer 2

numpyを使用しないで始めてください（あるいはnumpyを知らないかもしれません）。通常のpythonリストを使用してください。

x = [ 1, 2, 3, 4, 5 ] 
y = [ i * 2 if i < 3 else i + 2 for i in x ] 
print y 

出力：

[2, 4, 5, 6, 7] 

次に、あなたがそれnumpyの配列にすることができます。

x = np.array(x) 
y = np.array(y)