主页 / 问题 / 1402443
Accepted
Елена Сергеева
Asked:2022-06-21 18:49:09 +0000 UTC2022-06-21 18:49:09 +0000 UTC

如何将字符串切片转换为可用作数组切片的对象

  • 772

在一个循环中，我得到一个切片列表

qw = ['0:10, 0:20', '10:20, 20:40', '20:30, 40:60']

也就是说，每个切片都表示为一个字符串。之后我尝试将此切片应用于数组

im = [[[0,0,0], [0,0,0]], [[0,0,0], [0,0,0]], [[0,0,0], [0,0,0]]...]

但是由于 是一个字符串，那么它将不起作用。我试图这样做，但我意识到我会回到这条线上

ind = i.split(',')
ind = [int(i) for i in ind]
int_ind = ','.join(ind)

有必要通过给定的切片获取数组对象。

for i in qw:
  print(img[i])

我必须是一个正在运行的切片，即

img[0:10, 0:20]
python срез

1 个回答

  1. Best Answer

    1. 您需要从字符串中提取所有数字（这可以使用正则表达式来完成），将它们转换为整数

    2. 创建切片对象。这里的意思是当两个逗号分隔的切片是两个的元组时，视图切片img[a:b]是一个类型的对象。那些。：sliceslice

      • img[0:10]这与img[slice(0, 10)](或img[slice(0, 10, None)], 但空切片步骤可以省略)相同
      • img[0:10, 0:20]- img[(slice(0, 10), slice(0, 20))]。

      您可以使用以下代码验证这一点：

      class Test:
    def __getitem__(self, x):
        return x


print(Test()[0:10])  # slice(0, 10, None)
print(Test()[0:10, 0:20])  # (slice(0, 10, None), slice(0, 20, None))

    因此，您需要形成一个形式的对象，(slice(0, 10), slice(0, 20))从第一步开始替换数字。

    import numpy as np
import re


def convert_to_slice(s: str):
    a, b, c, d = map(int, re.findall(r"\d+", s))  # Вытаскиваем все числа из строки (считаем, что их всегда 4 штуки)
    return (slice(a, b), slice(c, d))  # Формируем объект нужного вида


img = np.array([[[0,0,0], [1,1,1]], [[2,2,2], [3,3,3]], [[4,4,4], [5,5,5]]])

# Результат должен получиться одинаковым
print(img[0:1, 0:2])
print(img[convert_to_slice('0:1, 0:2')])

    结果：

    [[[0 0 0]
  [1 1 1]]]
[[[0 0 0]
  [1 1 1]]]

    如果您查看上一个问题的上下文，那么您可以立即在循环中形成切片对象列表，而不是形成切片字符串：

    def convert_to_slice(a, b, c, d):
    return (slice(a, b), slice(c, d))


qw = []

for i in range(5):
    qw.append(convert_to_slice(
        i * y_shift, i * y_shift + y_size,
        i * x_shift, i * x_shift + x_size,
    ))
    • 3

相关问题