在Python 3中从较小的矩阵自动创建一个大矩阵

时间:2016-11-28 18:58:10

标签: python python-3.x numpy

我有许多较小的矩阵,我想将它们组合成一个更大的矩阵。

我试图在Numpy中使用hstack命令执行此操作,但它将每个组合视为字符串而不是字符串表示的矩阵。如果我手动编码,hstack命令工作正常。

import numpy as np 
import itertools as it 

sm01 = np.matrix( [[1],[1],[1]] ) 
sm02 = np.matrix( [[2],[2],[2]] )   
sm03 = np.matrix( [[3],[3],[3]] )   

Variables = ['sm01','sm02','sm03']
Comblst = ['']
for i in range(0, len(Variables)+1):
  for subset in it.combinations(Variables, i):
    a = str(subset).strip('()')
    Comblst = np.vstack([Comblst,a])

var = str(Comblst[6]).strip('[]').strip('""').strip(',').replace("'","")
var = tuple(var.split(', '))
LargeMatrix = np.hstack([var])

1 个答案:

答案 0 :(得分:0)

我认为您应该使用列表追加来构建Conblst,而不是vstack

Comblst = []
for
    for 
        Comblst.append(a)

向我们展示Comblstvar的样子。我只需阅读代码即可完成构建。

hstack的结果是什么。

====================

Out[1087]: ['sm01', 'sm02', 'sm03']
In [1088]: Comblst
Out[1088]: 
array([[''],
       [''],
       ["'sm01',"],
       ["'sm02',"],
       ["'sm03',"],
       ["'sm01', 'sm02'"],
       ["'sm01', 'sm03'"],
       ["'sm02', 'sm03'"],
       ["'sm01', 'sm02', 'sm03'"]], 
      dtype='<U22')

In [1090]: var = str(Comblst[6]).strip('[]').strip('""').strip(',').replace("'",
      ...: "")
      ...: var = tuple(var.split(', '))
      ...: 
In [1091]: var
Out[1091]: ('sm01', 'sm03')

In [1092]: np.hstack([var])
Out[1092]: 
array(['sm01', 'sm03'], 
      dtype='<U4')

这只是一堆字符串操作,使用列表而不是数组会更好。但是目的是什么?你真的打算生产

In [1093]: np.hstack([sm01, sm03])
Out[1093]: 
matrix([[1, 3],
        [1, 3],
        [1, 3]])

在Python中,字符串或名称以及它们引用的对象之间存在非常真实的区别。

我怀疑构建Comblst的方法更好:

In [1096]: 
      ...: Comblst = []
      ...: for i in range(0, len(Variables)+1):
      ...:   for subset in itertools.combinations(Variables, i):
      ...:     # a = str(subset).strip('()')
      ...:     Comblst.append(subset)
      ...:     
      ...:     
In [1097]: Comblst
Out[1097]: 
[(),
 ('sm01',),
 ('sm02',),
 ('sm03',),
 ('sm01', 'sm02'),
 ('sm01', 'sm03'),
 ('sm02', 'sm03'),
 ('sm01', 'sm02', 'sm03')]

In [1099]: Comblst[5]
Out[1099]: ('sm01', 'sm03')

忘记制作和分裂琴弦;而只是收集这些名字的元组。

但你不必使用这些名字;你可以从一个矩阵列表开始:

In [1101]: Mlist=[sm01, sm02, sm03]
In [1102]: 
      ...: Comblst = []
      ...: for i in range(0, len(Mlist)+1):
      ...:   for subset in itertools.combinations(Mlist, i):
      ...:     Comblst.append(subset)
      ...:     
In [1103]: Comblst
Out[1103]: 
[(), (matrix([[1],
          [1],
          [1]]),), (matrix([[2],
          [2],
          [2]]),), (matrix([[3],
          [3],
          ....
          [3]]))]
In [1104]: Comblst[5]
Out[1104]: 
(matrix([[1],
         [1],
         [1]]), matrix([[3],
         [3],
         [3]]))
In [1105]: np.hstack(Comblst[5])
Out[1105]: 
matrix([[1, 3],
        [1, 3],
        [1, 3]])

或者构造一个简单的数字元组列表,并用它构建索引

In [1109]: 
      ...: Comblst = []
      ...: for i in range(0, len(Mlist)+1):
      ...:   for subset in itertools.combinations( list(range(0, len(Mlist))), i):
      ...:     Comblst.append(subset)

In [1110]: Comblst
Out[1110]: [(), (0,), (1,), (2,), (0, 1), (0, 2), (1, 2), (0, 1, 2)]
In [1111]: Comblst[5]
Out[1111]: (0, 2)
In [1112]: np.hstack([Mlist[i] for i in Comblst[5]])
Out[1112]: 
matrix([[1, 3],
        [1, 3],
        [1, 3]])