目的:
一个朋友问我,未知行同时也无法获取行数的数据,对它按行进行全排列组合。
先整理一个简单的行向全排列代码:
n=3 #未知行
m=2 #列数
a=np.arange(m*n).reshape([n,m])
for i in range(2): #每行
c0=a[0,i]
for i in range(2):
c1=a[1,i]
for i in range(2):
c2=a[2,i]
print(c0,c1,c2)
效果:
[0, 2, 4]
[0, 2, 5]
[0, 3, 4]
[0, 3, 5]
[1, 2, 4]
[1, 2, 5]
[1, 3, 4]
[1, 3, 5]
上述代码主要是每层进行一个循环,循环套循环,最后一层循环终止,那么很简单。照着这个思路走:
import numpy as np
n=3 #未知行
m=2 #列数
a=np.arange(m*n).reshape([n,m])
c={}
c1={} #最后的结果
s=0
def ret1(x): #最后一层循环
global c,c1,s
for j in range(m): #列数
c[x]=a[x,j]
c1[s]=[c[i] for i in range(x+1)]
s=s+1
# for i in range(x+1): #测试一下
# c1.append(c[i])
# print(c1)
def ret(x):
global c
for j in range(m): #列数
try:
a[x+1,j] #错误跳转
c[x]=a[x,j]
except:
ret1(x) #当前x==n-1
break
xx=x+1
ret(xx)
ret(0)
效果:
[0, 2, 4]
[0, 2, 5]
[0, 3, 4]
[0, 3, 5]
[1, 2, 4]
[1, 2, 5]
[1, 3, 4]
[1, 3, 5]
朋友自己也想了个办法,附上他的代码:
import numpy as np
x = np.array([[1,2,3],[4,5,6],[7,8,9]])
def Amn(x):
from itertools import combinations
from functools import reduce
import numpy as np
import re
a = locals()
lenx = len(x)
dict1 = {}
for i in range(lenx):
a['list'+str(i+1)] = x[i]
dict1[str(i+1)] = a['list'+str(i+1)]
lists=[]
for key in combinations(dict1.keys(), lenx):
for i in key:
lists.append(dict1[i])
code = ','
fn = lambda x, code=',': reduce(lambda x,y: [str(i)+code+str(j) for i in x for j in y], x)
final = fn(lists, code)
ff = np.zeros((1,lenx))
for i in range(3**lenx):
temp = re.findall(r"-?\d+\.?\d*",str(final[i]))
cdr = np.zeros((1,lenx))
for j in range(lenx):
cdr[0,j] = float(temp[j])
ff = np.vstack((ff,cdr))
ff = np.delete(ff,0,axis=0)
return ff
ff = Amn(x)