如何将字符串形式的条件传入numpy where函数？

当然可以！你完全能通过字符串拼接的方式构建np.where所需的条件表达式，最直接的实现方式就是利用Python的eval()函数来解析字符串形式的布尔逻辑。下面我给你把示例补全，保证两种方式输出的结果完全一致：

import numpy as np

a = np.arange(10)
# 原始硬编码条件的方式
ind_original = np.where((a>6) | (a<3))
print("原始方式结果:", a[ind_original])

# 字符串动态构建条件的方式
select = 'c'
if select == 'c':
    sel0 = '(a>6)'
    sel = sel0 + ' | (a<3)'
    print("拼接后的条件字符串:", sel)
    # 用eval把字符串转换成可执行的布尔数组条件
    condition = eval(sel)
    ind_dynamic = np.where(condition)
    print("动态构建条件的结果:", a[ind_dynamic])

运行这段代码，你会看到两次输出的结果完全相同：[0 1 2 7 8 9]。

关键注意点 #

• 安全风险提示：eval()会执行传入的任意字符串代码，如果你的条件字符串来自不可信的外部输入（比如用户提交的内容），绝对不要用这种方式——这会带来严重的代码注入风险。
• 变量作用域问题：要确保字符串里用到的变量（比如这里的a）在eval()执行的当前作用域中存在，不然会抛出NameError。
• 更安全的替代方案：如果担心eval的安全问题，推荐用numpy内置的逻辑函数手动组合条件，比如np.logical_or()、np.logical_and()。你可以把每个条件片段封装成函数，再根据选择项组合它们的返回值：

import numpy as np

a = np.arange(10)

# 定义各个条件片段的函数
def cond_gt6(arr):
    return arr > 6

def cond_lt3(arr):
    return arr < 3

select = 'c'
if select == 'c':
    # 用numpy逻辑函数组合条件
    condition = np.logical_or(cond_gt6(a), cond_lt3(a))
    ind = np.where(condition)
    print(a[ind])

这种方式完全避免了eval的安全隐患，也更符合Python的最佳实践，适合在生产环境中使用。

内容的提问来源于stack exchange，提问作者VGP