我需要创建一个函数或方程式来转换此输入...
a = [True, False, True]
b = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
进入这个输出...
c = [[1, 3], [4, 6], [7, 9]]
注意:以上数组均为numpy数组。谢谢!
请您参考如下方法:
numpy解决方案
import numpy as np
a = np.array([True, False, True])
b = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
b[:,a]
array([[1, 3],
[4, 6],
[7, 9]])
