2016-07-05 59 views
3

我想從列表條目和一些額外的單個值創建一個數組。從列表和其他單獨值創建numpy數組的最佳方法

我用下面的方法似乎笨拙:

x=[1,2,3] 
y=some_variable1 
z=some_variable2 
x.append(y) 
x.append(z) 
arr = np.array(x) 

#print arr --> [1 2 3 some_variable1 some_variable2] 

有沒有更好的解決問題的辦法?

回答

3

您可以使用列表除了放置在列表中的所有變量添加到一個較大的,像這樣:

arr = np.array(x + [y, z]) 
+0

THX,這似乎比我的做法更清潔 – Dahlai

1

追加或串聯列表是罰款,也可能最快。

在陣列級級聯工程,以及

In [456]: np.hstack([x,y,z]) 
Out[456]: array([1, 2, 3, 4, 5]) 

這是緊湊,但在幕後確實

np.concatenate([np.array(x),np.array([y]),np.array([z])]) 
相關問題