数组问题

9,1
9,2
6,1565
6,1570
3,1
3,2
7,7
7,356
7,381
8,1734

有个二维数组，如上，第一列为分组ID，第二列为值，现在想实现添加一列，计算与各组的最大值的差。比如第一行，组ID为9，对应的最大值为2，故第一行更新后为9, 1, 1
同理第二行为 9, 2, 0

尽量别用暴力的方法，当然用个二重循环，很容易搞定

多键值排序不就好了？？

d = {}
for i in s.split('\n'):
        d.setdefault(i.split(',')[0],[]).append(i.split(',')[1])

for k,v in d.items():
        s = int(max(v))
        for i in v:
                print k+","+i+","+str(s-int(i))

二重循环是很容易搞定，不过我用的sql的方式搞定了

1. 
   
2. import pandas as pd
   
3. data=[[9,1       ],
   
4. [9,2       ],
   
5. [6,1565    ],
   
6. [6,1570    ],
   
7. [3,1       ],
   
8. [3,2       ],
   
9. [7,7       ],
   
10. [7,356     ],
    
11. [7,381     ],
    
12. [8,1734    ]]
    
13. df1 = pd.DataFrame(data)
    
14. df3=df1.join(df1.groupby(0).max(),0,'inner','l','r')
    
15. df3.columns=[0,1,2,3]
    
16. df3[4]=(df3[3]-df3[2])
    
17. print df3.iloc[:,[0,2,4]].values.tolist()
    

复制代码