python - 将条件从字符串传递到 numpy where

Question

我有一系列条件，这些条件可能会根据选择的名称而有所不同。我想知道是否有一种方法可以从一组字符串构造 python numpy where 的条件。下面有一个示例程序。我希望能够在 a[ind] 的两个实例中获得相同的结果。

import numpy as np

a = np.arange(10) 
ind = np.where((a>6) | (a<3))
print a[ind]

select = 'c'
if (select == 'c'):
    sel0 = '(a>6)'

sel = sel0 + ' | (a<3)'
print sel

ind = np.where(sel)
print a[ind]

这是我目前得到的输出:

[0 1 2 7 8 9]
(a>6) | (a<3)
[0]

Answer 1

您可以评估文字sel:

ind = np.where(eval(sel))
print(a[ind])
# [0 1 2 7 8 9]

<小时/>

但是为什么还要使用字符串文字呢？

select = 'c'
if select == 'c':
    sel0 = a>6

sel = sel0 | (a<3)
print(sel)

ind = np.where(sel)
print(a[ind])
# [0 1 2 7 8 9]

只要遵循相同的操作顺序，条件的计算位置实际上并不重要。无论您在何处评估条件，sel 将始终等于:

[ True  True  True False False False False  True  True  True]

使用该 bool 数组，where 将始终产生相同的输出。