在 NumPy 中使用字符串作为数组索引

Question

我正在通过 GUI 在 python 中处理大型数值数组。 我想将切片功能暴露给 GUI 中的文本框，这样我就可以轻松地选择应该用于手头计算的数组部分。

我想做的简单例子：

arr = array([0, 10, 20, 30, 40, 50, 60, 70, 80, 90])

a = "2:4" # example string from GUI Textbox
b = "[3, 4, 5]" # example string from GUI Textbox


print arr[a] # not valid code -> what should be written here to make it work?
print arr[b] # not valid code -> what should be written here to make it work?

应该输出：

[20, 30]
[30, 40, 50]

我发现了 slice 函数，但我需要手动解析我的字符串并创建一个 slice。有没有更简单的方法？

Answer 1

也许因为您只期望一个非常有限的字符集，所以使用eval 一次是可以接受的：

if not all(c in "1234567890-[],: " for c in b): # maybe also limit the length of b?
    # tell user you couldn't parse and exit this branch
slice_ = eval(f'np.s_[{b}]')
# slice_ can now be applied to your array: arr[slice_]

Answer 2

试试这个我认为它有效，假设你会在你的字符串中得到最小值和最大值。

import re
a = "2:4"
min = int(min(re.findall(r'(\d)',a)))
max = int(max(re.findall(r'(\d)',a)))
print arr[min:max]