返回子陣列

Question

考慮以下方便循環成語。

import numpy 

print "shape of" 
x = numpy.array([['a', 'b'], ['c', 'd']]) 
print x 
print "is", x.shape 
for row in x: 
    print "shape of", row, "is", row.shape 

這給

shape of 
[['a' 'b'] 
['c' 'd']] 
is (2, 2) 
shape of ['a' 'b'] is (2,) 
shape of ['c' 'd'] is (2,) 

我的問題是，一個可以保存方便for row in x成語在返回其具有形狀（2,1）的陣列，在這種情況下？謝謝。 將子陣列的形狀從（2，）轉換爲（2,0）的函數會很好。例如。

for row in x: 
    print "shape of", somefunc(row), "is", row.shape 

返回

shape of ['a' 'b'] is (2,1) 

Answer 1

我不明白你爲什麼會想這一點，但你可以試試這個：

for row in x: 
    print "shape of", row, "is", numpy.reshape(row, (1, row.size)).shape 

在我optinion，一維數組更容易處理。所以把它重新塑造成「1d矩陣」並沒有什麼意義。

Answer 2

您可以使用numpy.expand_dim增加任何numpy的陣列的排名：

In [7]: x=np.array([1,2,3,4]) 

In [8]: x.shape 
Out[8]: (4,) 

In [9]: np.expand_dims(x,axis=-1).shape 
Out[9]: (4, 1)