2017-04-10 114 views
1

我有一些python的问题。蟒蛇并排追加两个矩阵

A= [ [1,2,3] 
     [4,5,6] 
     ] 

    B = [ [10,11] 
     [12,13] 
     ] 

我想有:

C = [A B] 
    = [ [1,2,3, 10, 11] 
     [4,5,6, 12, 13] 
     ] 

我该怎么做,在蟒蛇?

回答

4

您正在描述基本用法np.hstack

np.hstack((A, B)) 

还有一个等效指标的绝招:

np.c_[A, B] 
2

你可以做这样的事情,在压缩基本将每个列表(A,B)对象:

>>> [x + y for x, y in zip(A, B)] 
[[1, 2, 3, 10, 11], [4, 5, 6, 12, 13]] 
3

另外,使用np.concatenateaxis=1

In [207]: np.concatenate((A, B), axis=1) 
Out[207]: 
array([[ 1, 2, 3, 10, 11], 
     [ 4, 5, 6, 12, 13]]) 

如果你关心性能,np.concatenate是真正的战争马。

In [215]: %timeit np.concatenate((A, B), 1) 
The slowest run took 12.10 times longer than the fastest. 
100000 loops, best of 3: 3.1 µs per loop 

In [214]: %timeit np.hstack((A,B)) 
The slowest run took 6.85 times longer than the fastest. 
100000 loops, best of 3: 12.5 µs per loop 

In [216]: %timeit np.c_[A, B] 
10000 loops, best of 3: 48.7 µs per loop 
如果你问列表的numpy的阵列(按你的标签)或列表(按你的标题矩阵)
+1

也许是值得一提的是' np.hstack'实际上使用'np.concatenate',所以它永远不会更快。我们应该更喜欢使用更高级的API,除非最大性能是必要的。 – wim

1

不能告诉...但:

来连接列表中,您只需将它们添加一起:

A = [[1,2,3],[4,5,6]] 
B = [[10,11],[12,13]] 
C = [A[0]+B[0],A[1]+B[1]] 

print C 

>>> 
[[1, 2, 3, 10, 11], [4, 5, 6, 12, 13]] 

,如果你有阵列,你开始可以使用append方法:

import numpy as np 
A = np.array([[1,2,3],[4,5,6]]) 
B = np.array([[10,11],[12,13]]) 
C = np.array([np.append(A[0],B[0]),np.append(A[1],B[1])]) 

print C 

>>> 
[[ 1 2 3 10 11] 
[ 4 5 6 12 13]]