Python numpy.concatenate()

简介

numpy.concatenate() 是 numpy 库中的一个函数，用于沿特定轴连接两个或多个数组。它可以在指定的轴上将多个数组沿着该轴连接起来，从而创建一个更大的数组。

语法

numpy.concatenate((array1, array2, ...), axis=0, out=None)

参数:

• array1, array2, ...：需要连接的数组。
• axis：可选参数，指定连接的轴。默认为 0，表示沿着第一个轴连接。如果 axis 为 None，则数组会被展开。
• out：可选参数，指定输出数组。

返回值:

• 连接后的新数组。

示例

import numpy as np

# 连接两个一维数组
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
result = np.concatenate((arr1, arr2))
print(result)
# 输出: [1 2 3 4 5 6]

# 连接两个二维数组
arr3 = np.array([[1, 2], [3, 4]])
arr4 = np.array([[5, 6]])
result = np.concatenate((arr3, arr4), axis=0)
print(result)
# 输出: [[1 2]
#        [3 4]
#        [5 6]]

# 沿列连接两个二维数组
result = np.concatenate((arr3, arr4), axis=1)
print(result)
# 输出: [[1 2 5]
#        [3 4 6]]

说明

• numpy.concatenate() 可以用于连接多个数组，使数组维度相同。
• 沿着指定的轴连接数组时，需要确保其他轴的长度相同，否则会抛出 ValueError 异常。

详细的信息和示例可以参考 numpy.concatenate()文档。