Nối hai mảng NumPy một chiều

Tôi có hai mảng một chiều đơn giản trong NumPy . Tôi có thể nối chúng bằng cách sử dụng numpy.concatenate . Nhưng tôi nhận được lỗi này cho mã dưới đây:

TypeError: chỉ các mảng có độ dài-1 có thể được chuyển đổi sang vô hướng Python

Mã

import numpy
a = numpy.array([1, 2, 3])
b = numpy.array([5, 6])
numpy.concatenate(a, b)

Tại sao?

Dòng nên là:

numpy.concatenate([a,b])

Các mảng bạn muốn nối cần phải truyền vào dưới dạng một chuỗi, không phải là các đối số riêng biệt.

Từ tài liệu NumPy :

numpy.concatenate((a1, a2, ...), axis=0)

Tham gia một chuỗi các mảng với nhau.

Nó đã cố gắng diễn giải bthông số trục của bạn , đó là lý do tại sao nó phàn nàn rằng nó không thể chuyển đổi nó thành vô hướng.

Có một số khả năng để ghép các mảng 1D, ví dụ:

numpy.r_[a, a],
numpy.stack([a, a]).reshape(-1),
numpy.hstack([a, a]),
numpy.concatenate([a, a])

Tất cả các tùy chọn đều nhanh như nhau cho các mảng lớn; Đối với những người nhỏ, concatenatecó một cạnh nhẹ:

Cốt truyện được tạo ra với perfplot :

import numpy
import perfplot

perfplot.show(
    setup=lambda n: numpy.random.rand(n),
    kernels=[
        lambda a: numpy.r_[a, a],
        lambda a: numpy.stack([a, a]).reshape(-1),
        lambda a: numpy.hstack([a, a]),
        lambda a: numpy.concatenate([a, a]),
    ],
    labels=["r_", "stack+reshape", "hstack", "concatenate"],
    n_range=[2 ** k for k in range(19)],
    xlabel="len(a)",
)

Tham số đầu tiên concatenatephải là một chuỗi các mảng để nối:

numpy.concatenate((a,b)) # Note the extra parentheses.

Một ist khác để sử dụng dạng "concatenate" ngắn là "r _ [...]" hoặc "c _ [...]" như được hiển thị trong mã ví dụ bên dưới (xem http://wiki.scipy.org / NumPy_for_Matlab_Users để biết thêm thông tin):

%pylab
vector_a = r_[0.:10.] #short form of "arange"
vector_b = array([1,1,1,1])
vector_c = r_[vector_a,vector_b]
print vector_a
print vector_b
print vector_c, '\n\n'

a = ones((3,4))*4
print a, '\n'
c = array([1,1,1])
b = c_[a,c]
print b, '\n\n'

a = ones((4,3))*4
print a, '\n'
c = array([[1,1,1]])
b = r_[a,c]
print b

print type(vector_b)

Kết quả nào trong:

[ 0.  1.  2.  3.  4.  5.  6.  7.  8.  9.]
[1 1 1 1]
[ 0.  1.  2.  3.  4.  5.  6.  7.  8.  9.  1.  1.  1.  1.] 


[[ 4.  4.  4.  4.]
 [ 4.  4.  4.  4.]
 [ 4.  4.  4.  4.]] 

[[ 4.  4.  4.  4.  1.]
 [ 4.  4.  4.  4.  1.]
 [ 4.  4.  4.  4.  1.]] 


[[ 4.  4.  4.]
 [ 4.  4.  4.]
 [ 4.  4.  4.]
 [ 4.  4.  4.]] 

[[ 4.  4.  4.]
 [ 4.  4.  4.]
 [ 4.  4.  4.]
 [ 4.  4.  4.]
 [ 1.  1.  1.]]

Dưới đây là phương pháp tiếp cận để thực hiện điều này bằng cách sử dụng numpy.ravel(), numpy.array(), sử dụng thực tế là mảng 1D có thể giải nén vào các yếu tố đơn giản:

# we'll utilize the concept of unpacking
In [15]: (*a, *b)
Out[15]: (1, 2, 3, 5, 6)

# using `numpy.ravel()`
In [14]: np.ravel((*a, *b))
Out[14]: array([1, 2, 3, 5, 6])

# wrap the unpacked elements in `numpy.array()`
In [16]: np.array((*a, *b))
Out[16]: array([1, 2, 3, 5, 6])

Một số sự kiện khác từ các tài liệu numpy :

Với cú pháp như numpy.concatenate((a1, a2, ...), axis=0, out=None)

trục = 0 đối với trục nối theo hàng-khôn ngoan = 1 đối với ghép nối theo cột

>>> a = np.array([[1, 2], [3, 4]])
>>> b = np.array([[5, 6]])

# Appending below last row
>>> np.concatenate((a, b), axis=0)
array([[1, 2],
       [3, 4],
       [5, 6]])

# Appending after last column
>>> np.concatenate((a, b.T), axis=1)    # Notice the transpose
array([[1, 2, 5],
       [3, 4, 6]])

# Flattening the final array
>>> np.concatenate((a, b), axis=None)
array([1, 2, 3, 4, 5, 6])

Tôi hy vọng nó sẽ giúp!