np.hstack()
是把矩阵进行行连接。
np.vstack()
是把矩阵进行列连接。
行连接np.hstack()
代码示例①:
import numpy as np
a=np.array([1,2,3])
b=np.array([2,3,4])
aa=[1,2,3] # 列表也可以作为参数传入
bb=[2,3,4]
print("np.hstack((a,b)):",type(np.hstack((a,b))),"\n",np.hstack((a,b)))
print("np.hstack((a,b)):",type(np.hstack((aa,bb))),"\n",np.hstack((aa,bb)))
c=np.array([[1],
[2],
[3]])
d=np.array([[2],
[3],
[4]])
print("np.hstack((c,d)):\n",np.hstack((c,d)))
np.hstack((a,b)): <class 'numpy.ndarray'>
[1 2 3 2 3 4]
np.hstack((a,b)): <class 'numpy.ndarray'>
[1 2 3 2 3 4]
np.hstack((c,d)):
[[1 2]
[2 3]
[3 4]]
行连接np.hstack()
代码示例②:
import numpy as np
s=[1,1,1,1]
a=2
r=3
s_=[4,4,4,4]
output = np.hstack((s, [a, r], s_))
print(output)
[1 1 1 1 2 3 4 4 4 4]
同理列连接np.vstack()
也类似
列连接np.vstack()
代码示例:以下代码仅仅把hstack
改成了vstack
。(在pycharm中可用ctrl+r
批量替换)
import numpy as np
a=np.array([1,2,3])
b=np.array([2,3,4])
aa=[1,2,3] # 列表也可以作为参数传入
bb=[2,3,4]
print("np.vstack((a,b)):",type(np.vstack((a,b))),"\n",np.vstack((a,b)))
print("np.vstack((a,b)):",type(np.vstack((aa,bb))),"\n",np.vstack((aa,bb)))
c=np.array([[1],
[2],
[3]])
d=np.array([[2],
[3],
[4]])
print("np.vstack((c,d)):\n",np.vstack((c,d)))
np.vstack((a,b)): <class 'numpy.ndarray'>
[[1 2 3]
[2 3 4]]
np.vstack((a,b)): <class 'numpy.ndarray'>
[[1 2 3]
[2 3 4]]
np.vstack((c,d)):
[[1]
[2]
[3]
[2]
[3]
[4]]