python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > numpy 拼接矩阵

numpy拼接矩阵的实现

作者:头戴面包圈

本文主要介绍了numpy拼接矩阵的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧

1、文档

使用numpy的 concatenate 拼接矩阵,文档里面这样解释:

numpy.concatenate((a1, a2, ...), axis=0, out=None, dtype=None, casting="same_kind")

(a1, a2, ...):连接的数组必须有一样的维度;

axis:拼接的方向;

out:预设输出矩阵的大小

…………

2、举例

首先给定两个矩阵:

rotation = np.array([[1, 2, 3],
                     [4, 5, 6],
                     [7, 8, 9]])
 
trans = np.array([[7],
                  [8],
                  [0]])

①:在第一个矩阵后面加上第二个矩阵(加一列):

z = np.concatenate((rotation, trans), axis=1)

输出为:

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

②:以及在矩阵下面加一个全零行:

add_arr = np.array([[0, 0, 0, 0]])
array_by_add = np.concatenate((z, add_arr), axis=0)

输出:

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

③:把矩阵填充为对角矩阵:

第一个为3×3矩阵,第二个为2×3矩阵。

arr1 = np.array([[2, 3, 4],
                 [3, 4, 5],
                 [4, 5, 6]])
arr2 = np.array([[7, 8, 9],
                 [8, 9, 10]])
 
arr_zeros1 = np.zeros((arr2.shape[1], arr1.shape[0]))
print(arr_zeros1)
arr_zeros2 = np.zeros((arr2.shape[0], arr1.shape[1]))
print(arr_zeros2)
total_arr1 = np.concatenate((arr1, arr_zeros1), axis=1)
total_arr2 = np.concatenate((arr_zeros2, arr2,), axis=1)
total_arr = np.concatenate((total_arr1, total_arr2), axis=0)
print(total_arr)

拼接之后结果如下:

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

官方文档地址:numpy官网地址

到此这篇关于numpy拼接矩阵的实现的文章就介绍到这了,更多相关numpy 拼接矩阵内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文