如何在Python中将嵌套元组的矩阵拆分为块？

2024-09-08 • 问答

问题在于如何将矩阵拆分为块。

如果我有4x4矩阵，则如下所示。

[[a,b,c,d],[e,f,g,h],[h,i,j,k],[l,m,n,o]]

我想分成以下几块：

[a,b]  
[c,d]  

[h,i]  
[j,k]

[e,f]  
[g,h]  

[l,m]  
[n,o]

但是a，b，c ... o是嵌套的元组，例如a = [(x1,y1),(x2,y2),...(xn,yn)]。 n不一定等于矩阵中的每个元素。

是否有任何内置函数可以做到这一点，或者我必须操纵索引并一一挑选元素？我尝试使用np.vsplit()和np.hsplit()，但它们对我的数据无效。

更新： np.vsplit()和np.hsplit()可能会解决我的问题，但还有一个问题。我得到的数据是16个坐标列表，但是这些列表的长度（每个列表中2d坐标的数量）不相等。为了使用np.vsplit()和np.hsplit()，我必须将16个列表转换为numpy ndarray并对其进行整形。此ndarray的形状应类似于(4,4,x,2)，这意味着它们被放置在4行，4列中，每个元素都有x=number_of_coordinates_in_that_list和2列。我可以使用np.reshape()实现我想要的吗？

Data = [ [(),(),...],[(),... ] # Calculate block start index(Upper left). rows = int(matrix_size / block_size) columns = int(matrix_size / block_size) index_0s = [] for row in range(rows): for column in range(columns): index_0s.append(row * matrix_size * block_size + column * block_size) def generate_block(block_start_index): indices = [] for i in range(block_size): for j in range(block_size): index_ij = matrix_size * i + block_start_index + j indices.append(index_ij) block = [] for block_index in indices: block.append(Data[block_index]) return block # Generate all blocks. blocks = [] for index_0 in index_0s: block_i = generate_block(index_0) blocks.append(block_i)

如何在Python中将嵌套元组的矩阵拆分为块？

shanghaiditu 回答：如何在Python中将嵌套元组的矩阵拆分为块？

大家都在问