Numpy的array合并:

import numpy as np

a = np.array([1, 1, 1])
b = np.array([2, 2, 2])

c = np.vstack((a, b))  # vertical stack
d = np.hstack((a, b))  # horizontal stack
print(c)
print(d)
print(c.shape, d.shape)
"""
[[1 1 1]
 [2 2 2]]
 [1 1 1 2 2 2]
(2, 3) (6,)
"""

print(a[np.newaxis, :])  # [[1 1 1]]
print(a[:, np.newaxis])
"""
[[1]
 [1]
 [1]]
"""

e = np.concatenate((a, b, b, a), axis=0)
print(e)
# [1 1 1 2 2 2 2 2 2 1 1 1]
最后修改:2019 年 05 月 24 日 01 : 05 PM
如果觉得我的文章对你有用,请随意赞赏