python - 在python中连接几个np数组

Question

我有几个颠簸的数组，我想连接它们。我正在使用 np.concatenate((array1,array2),axis=1)。我现在的问题是我想让数组的数量参数化，我写了这个函数

x1=np.array([1,0,1])
x2=np.array([0,0,1])
x3=np.array([1,1,1])  

def conc_func(*args):
    xt=[]
    for a in args:
        xt=np.concatenate(a,axis=1)
    print xt
    return xt

xt=conc_func(x1,x2,x3)

这个函数返回([1,1,1])，我希望它返回([1,0,1,0,0,1,1,1,1])。我试图在 np.concatenate 中添加 for 循环

xt =np.concatenate((for a in args: a),axis=1)

但我遇到语法错误。我既不能使用 append 也不能使用 extend，因为我必须处理 numpy arrays 而不是 lists。有人可以帮忙吗？

提前致谢

Answer 1

concatenate可以接受一系列类似数组的序列，例如 args:

In [11]: args = (x1, x2, x3)

In [12]: xt = np.concatenate(args)

In [13]: xt
Out[13]: array([1, 0, 1, 0, 0, 1, 1, 1, 1])

顺便说一下，虽然 axis=1 有效，但输入都是一维数组(因此它们只有一个 0 轴)。因此，使用 axis=0 或完全省略 axis 更有意义，因为默认值为 axis=0。