numpy配列内のいくつかの列項目を返す方法は？

Question

Iは、例えば、2D numpyの配列にいくつかのアイテムを印刷したい：

a = [[1, 2, 3, 4], 
    [5, 6, 7, 8]] 

a = numpy.array(a) 

どのように（図5、図6）、また（1及び2）だけ戻り、そしてどのように[ように寸法を維持することができ2,2] ??

Answer 1

次

a[:, [0, 1]] 

は、最初の2つの列を選択する（インデックス0と1）。結果は次のようになります。

array([[1, 2], 
     [5, 6]]) 

Answer 2

numpyの配列の必要な部分を取得するためにスライシングを使用できます。例えば、あなたはまた、2x2の部分配列を選択することができます5と6

>>> a[1, 0:2] 
array([5, 6]) 

のために同様に1と2あなたが0の行を選択する必要があり、最初の2つの列、すなわち

>>> a[0, 0:2] 
array([1, 2]) 

を取得するには

>>> a[:,0:2] 
array([[1, 2], 
     [5, 6]]) 

Answer 3

あなたはこのように行うことができ、

In [44]: a[:, :2] 
Out[44]: 
array([[1, 2], 
     [5, 6]])