Python矩阵,交替拼接
我手上有两个python的numpy矩阵,A矩阵和B矩阵,两者大小一样,都是(4096,),我想把他们拼接到一起,得到C矩阵,C矩阵的偶数行等于A矩阵,C矩阵的奇数行等于B矩阵;
要实现你的需求,即创建一个新的矩阵C
,其中偶数行来自A
矩阵,奇数行来自B
矩阵,且考虑到A
和B
都是一维数组(形状为(4096,)
),我们需要首先将它们转换为二维数组(即矩阵),以便能够按照行进行拼接。不过,由于你的原始数组实际上是一维的,我们可以通过调整索引和重塑(reshape)来实现目标,但最终的“矩阵”概念可能有些模糊,因为在实际操作中,我们更多是在处理一维数组的交错排列。
不过,为了符合你的描述,我们可以先假设你想要的C
是一个二维数组,其形状为(8192, 1)
(因为你有4096个偶数行和4096个奇数行),但实际上每个“行”只是一个元素。但更常见的是,我们可能会将结果视为一个一维数组,其中元素交替来自A
和B
。
这里我提供一个将A
和B
交错拼接成一维数组的方法,同时也提供一个将其视为二维数组(8192x1
)的方法:
一维数组交错拼接
import numpy as np
# 假设A和B是两个一维numpy数组,形状为(4096,)
A = np.arange(4096) # 示例数组A
B = A + 4096 # 示例数组B,为了区分,这里简单地让B的每个元素比A大4096
# 交错拼接A和B
C = np.empty(8192, dtype=A.dtype)
C[0::2] = A # 偶数索引位置放置A的元素
C[1::2] = B # 奇数索引位置放置B的元素
# 此时C是一个一维数组,形状为(8192,)
print(C.shape) # 输出: (8192,)
二维数组(8192x1)
如果你真的需要一个二维数组,其中每“行”是一个元素,你可以这样做:
# 将C重塑为二维数组(8192x1)
C_2d = C.reshape(8192, 1)
# 此时C_2d是一个二维数组,形状为(8192, 1)
print(C_2d.shape) # 输出: (8192, 1)
但请注意,这种二维表示在实际应用中可能并不常见,因为它并没有真正利用二维数组的优势(即按行和列组织数据)。在大多数情况下,你可能会直接使用上面创建的一维数组C
。