計算矩陣中不同列的數量

Question

我有一些簡單的代碼，我只想計算兩個矩陣乘積中不同列的數量。該代碼是

import numpy as np 
import itertools 

n = 5 
h = 2 
M = np.random.randint(2, size=(h,n)) 
F = np.matrix(list(itertools.product([0,1],repeat = 5))).transpose() 
product = M*F 
setofcols = set() 
for column in product.T: 
    setofcols.add(column) 
print len(setofcols) 

然而，這給出了錯誤的答案爲setofcols所有元素都是即使列是相同的不同。什麼是正確的做法？

我會用更大的n和h值來運行它，所以儘可能少使用內存的解決方案會很好。

Answer 1

你可以讓你的工作，通過使用repr：

import numpy as np 
import itertools 

n = 5 
h = 2 
M = np.random.randint(2, size=(h,n)) 
F = np.matrix(list(itertools.product([0,1],repeat = 5))).transpose() 
product = M*F 
setofcols = set() 
for column in product.T: 
    setofcols.add(repr(column)) 
print len(setofcols) 
print setofcols 

你也可以這樣做：

setofcols={tuple(e.A1) for e in product.T} 

凡矩陣的A1屬性是1D基本陣列可以用來作爲tuple的序列。