如果标题不是很清楚，很抱歉.我在构思这个问题上有些困难.
我有两个N×M维的数字array.为简单起见，假设它们都有形状(2，10).first数组由浮点数组成.例如:
[[0.1,0.02,0.2,0.3,0.013,0.7,0.7,0.11,0.18,0.6],
[0.23,0.02,0.1,0.1,0.011,0.3,0.4,0.4,0.4,0.5]]
百位数组由0‘S和1’S组成，例如:
[[0,1,0,0,0,0,1,1,0,1],
[1,0,1,0,0,0,0,1,1,0]]
我正在try 执行以下操作:对于第二个数组中唯一值的给定配置， Select 第一个数组中该位置的元素.举个例子，我们在第二个数组中有两行，所以1和0有4种可能的配置.即(1，1)，(0，0)，(1，0)，(0，1).如果我们采用(1，1)的情况(即，第1行和第2行中的元素都等于第二个数组中的'1')，我想找到这些值并查找它们在第一个数组中的位置.这将从第一个数组返回(0.11，0.4).
再次，如果没有清楚地传达这一点，我向您道歉.感谢您的任何反馈.谢谢.
IIUC，您想要搜索列中arr2中1都是1的位置上的arr1:
arr1 = np.array(
[
[0.1, 0.02, 0.2, 0.3, 0.013, 0.7, 0.7, 0.11, 0.18, 0.6],
[0.23, 0.02, 0.1, 0.1, 0.011, 0.3, 0.4, 0.4, 0.4, 0.5],
]
)
arr2 = np.array([[0, 1, 0, 0, 0, 0, 1, 1, 0, 1], [1, 0, 1, 0, 0, 0, 0, 1, 1, 0]])
out = arr1[:, np.all(arr2, axis=0)]
print(out)
打印:
[[0.11]
[0.4 ]]
如果要查找所有组合:
unique = np.unique(arr2.T, axis=0)
for row in unique:
print("Combination:")
print(row)
print()
print(arr1[:, np.all(arr2 == row.reshape(arr2.shape[0], -1), axis=0)])
print("-" * 80)
打印:
Combination:
[0 0]
[[0.3 0.013 0.7 ]
[0.1 0.011 0.3 ]]
--------------------------------------------------------------------------------
Combination:
[0 1]
[[0.1 0.2 0.18]
[0.23 0.1 0.4 ]]
--------------------------------------------------------------------------------
Combination:
[1 0]
[[0.02 0.7 0.6 ]
[0.02 0.4 0.5 ]]
--------------------------------------------------------------------------------
Combination:
[1 1]
[[0.11]
[0.4 ]]
--------------------------------------------------------------------------------