2017-02-02 74 views
2

假設您想要創建2D numpy數組,並且想用順序整數填充它。 有什麼選擇?Numpy:使用順序整數填充2D numpy數組

例子:

import numpy 
a=numpy.array([[0,1,2,3,4],[5,6,7,8,9],[10,11,12,13,14]]) 

在這種情況下,我當然可以使用range(15)確定上限,但如何告訴numpy5值來創建一個新的行?我正在尋找一個通用解決方案,因爲我真正的問題圍繞着一個有466列和365行的二維數組。

+3

你可以做一個'= np.arange(15).reshape(3,5)'爲您簡單的例子,對於一般情況下'A = np.arange(466 * 365).reshape(466365 )'應該工作 – EdChum

回答

5

只需在正常的np.arange調用上使用.reshape,該調用將簡單地重新整形陣列而不更改其數據。

In[101]: np.arange(15).reshape(3, 5) 
Out[101]: 
array([[ 0, 1, 2, 3, 4], 
     [ 5, 6, 7, 8, 9], 
     [10, 11, 12, 13, 14]]) 

In[102]: np.arange(466*365).reshape(466, 365) 
Out[102]: 
array([[  0,  1,  2, ..., 362, 363, 364], 
     [ 365, 366, 367, ..., 727, 728, 729], 
     [ 730, 731, 732, ..., 1092, 1093, 1094], 
     ..., 
     [168995, 168996, 168997, ..., 169357, 169358, 169359], 
     [169360, 169361, 169362, ..., 169722, 169723, 169724], 
     [169725, 169726, 169727, ..., 170087, 170088, 170089]])