弱鸡求助！一个复杂匹配功能如何实现！

akaduoduo

求助鱼C的大佬们！
有如下Dataframe：
Dr  Cr
a  10   0
b   0   1
c   0   6
d   2   0
e   3   0
f   7   0
g   0   5
h   0   2
i   0   3
j   0   5

想实现的效果是（针对非零数字）：
1. 找到Dr列和Cr列中一样的数字，输出相关数字所在行：
d   2   0
h   0   2

e   3   0
i   0   3
2. 找到Cr列中几个数字加起来等于Dr列中的某个数字的所有组合：
a  10   0
g   0   5
j   0   5

a  10   0
b   0   1
c   0   6
i   0   3

f   7   0
g   0   5
h   0   2

f   7   0
h   0   2
j   0   5

没有穷举，类似是这样的效果...

关于第1个功能，我自己的思路如下：

1. for i in list(df['Cr']):
   
2.     if i == 0:
   
3.         continue
   
4.             
   
5.     if i in list(df['Dr']):
   
6.         df1 = df.iloc[np.flatnonzero(df['Dr'] == i)]
   
7.         df2 = df.iloc[np.flatnonzero(df['Cr'] == i)]
   
8.         df = pd.concat([df1,df2])
   
9.         print(df)

复制代码

但是结果如下，不知道为何只显示出第1个匹配项：
   Dr  Cr
d   2   0
h   0   2

关于第2个功能...毫无头绪...
360度后空翻跪地磕头求助！！！

塔利班

第一个问题是你把新生成的df给df自己了，那后续能遍历出来个啥？
后面的先研究差值，记得去重，递归下就行，还有0能被各种0加起来不算么

akaduoduo

塔利班 发表于 2019-12-10 18:12
第一个问题是你把新生成的df给df自己了，那后续能遍历出来个啥？
后面的先研究差值，记得去重，递归下就行 ...

谢谢大佬指点，第1个问题解决了。。。

关于第2个问题，能再请教下递归具体怎么写吗？
我现在仅有一点思路是笨办法用itertools，把Cr列分成2个一组求和，再到Dr列匹配：

1. list_cr = list(df['Cr'])
   
2. comb2 = itertools.combinations(list_cr, 2)
   
3. result2 = [sum(i) for i in comb2]
   
4. 
   
5. for i in result2:
   
6.     if i == 0:
   
7.         continue
   
8. 
   
9.     if i in list(df['Dr']):
   
10.         print(df.iloc[np.flatnonzero(df['Dr'] == i)])

复制代码

但是不知道要怎么继续按3个一组、4个一组、5个一组这样循环下去...

塔利班

先统计没0的，这个可以去看leetocde n数之和类似的,我最近太忙，没时间写这样的代码- -，除非RMB