如何在给定条件下选择数组的元素？
假设我有一个numpy数组 x = [5, 2, 3, 1, 4, 5] 和 y = ['f', 'o', 'o', 'b', 'a', 'r']。我想选择在x中大于1且小于5的元素所对应的y中的元素。
我尝试了以下代码
x = array([5, 2, 3, 1, 4, 5])
y = array(['f','o','o','b','a','r'])
output = y[x > 1 & x < 5] # 期望的输出是 ['o','o','a']
但是这个方法不起作用。我该怎么做呢？
