假设我们有一个 100*4 的数组。
我们还有一个 1 和 0 的 100*1 数组。假设有 n 个 1。
我们想从 100*4 数组创建一个*4 数组,其中我们只包含第二个数组为 1 的列。
一种方法是通过双 for 循环。有没有更简单的方法?
所以,我们有
A = [ [ 332 44 33 22 33 55 33 211 .....
[ 823 44 12 98 19 23 32 911 .....
....
....
]
和
B = [1 0 0 1 0 0 0 ....]
我们想要
C = [ [ 332 22 ...
[ 823 98 ...
....
....
]