[Numpy] concatenate

numpy.concatenate

接受任意多个数组作为参数，可以指定指定轴，默认为0。

import numpy as np
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6]])
print(np.concatenate((a, b), axis=0))
print(np.concatenate((a, b)))
print(np.concatenate((a, b.T), axis=1))

[[1 2]
 [3 4]
 [5 6]]
[[1 2]
 [3 4]
 [5 6]]
[[1 2 5]
 [3 4 6]]

对于MaskedArray，不会保存mask的信息，除非使用numpy.ma.concatenate()

import numpy as np
a = np.ma.arange(3)
a[1] = np.ma.masked
b = np.arange(2, 5)
print(a)
print(b)
print(np.concatenate([a, b]))
print(np.ma.concatenate([a, b]))

[0 -- 2]
[2 3 4]
[0 1 2 2 3 4]
[0 -- 2 2 3 4]