no | industry | group | assets | g1 |
1 | A | 1 | 100 | 1 |
2 | B | 4 | 90 | . |
3 | A | 1 | 30 | 1 |
4 | C | 1 | 70 | . |
5 | C | 3 | 10 | . |
6 | B | 2 | 50 | . |
7 | C | 2 | 40 | . |
8 | A | 3 | 60 | 3 |
9 | B | 4 | 20 | . |
数据如图 仅列了10家公司,no表示代码,industry表示行业已经encode对应了数字,group表示集团编码,g1是“当industry为A时,g1=group”。另外生成g1[3*1]的矩阵。
现在想要做的是一个关于矩阵的循环:当group里面的值等于矩阵中 r1 或 r2 或 r3 ------rn,则求所有符合条件的公司assets之和 以及均值等。