针对未知行数据全排列组合的办法

目的:
一个朋友问我,未知行同时也无法获取行数的数据,对它按行进行全排列组合。

先整理一个简单的行向全排列代码:

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)
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值