数组和字符串操作

Question

我有两个数组：（arrayA 包含整数，arrayB 和 arrayC 包含字符串）

arrayA = [5, 4]
arrayB = ["id_1", "id_2"] #arrayA and arrayB always have same lengths 
arrayC = ['3', '4', '15', '20', '23', '8', '11', '14', '21']

我想要结果：

res = {"id_1": ['3', '4', '15', '20', '23'], 
       "id_2": ['8', '11', '14', '21']}

所以基本上我想告诉程序 arrayB 的 id_1 包含前 5 个字符串，而 arrayB 的 id_2 包含接下来的 4 个字符串。

类似以下问题的结果是：

A = [2, 1, 3, 4]
B = ["a", "b", "c", "d"]
C = ['23', '8', '11', '3', '4', '15', '20', '23', '100', '45'] 

res = {"a": ['23', '8'], 
       "b": ['11'],
       "c": ['3', '4', '15'], 
       "d": ['20', '23', '100', '45']}

Answer 1

您可以根据A 提供的长度，使用itertools.islice 剥离C 的部分。为此，您还需要将 C 设为迭代器，以便切片从最后一个切片停止的位置开始：

import itertools as IT
A = [2, 1, 3, 4]
B = ["a", "b", "c", "d"]
C = ['23', '8', '11', '3', '4', '15', '20', '23', '100', '45'] 
C = iter(C)
result = {bi: list(IT.islice(C, ai)) for ai, bi in zip(A, B)}
print(result)

产量

{'b': ['11'], 
 'c': ['3', '4', '15'], 
 'd': ['20', '23', '100', '45'], 
 'a': ['23', '8']}

Answer 2

你可以试试：

A = [2, 1, 3, 4]
B = ["a", "b", "c", "d"]
C = ['23', '8', '11', '3', '4', '15', '20', '23', '100', '45'] 


res = {}
ix = 0
for k, l in zip(B, A):
    res[k] = C[ix:(ix+l)]
    ix += l

print(res)

# {'a': ['23', '8'], 'b': ['11'], 'c': ['3', '4', '15'], 'd': ['20', '23', '100', '45']}

解决方案如下：

首先，我们创建一个名为res 的字典来跟踪结果。然后，我们创建一个名为 ix 的变量并将其设置为 0。该变量帮助我们跟踪数组 C 的索引位置，以便为我们的结果字典获取所需的值。此外，我们zip 数组B 和A。这个压缩过程相当于从B 和A 创建一个新的元组列表，看起来像[("a", 2), ("b", 1), ("c", 3), ("d", 4)]。现在您已经有了这个压缩容器，我们使用 for-loop 进行迭代；因此for k, l in zip(B, A) 部分。随后，在每次迭代中，我们将数组 C 从 ix 切片到 ix + l；其中l 是数组A 中对应的整数值。此切片操作的值保存在我们的 res 字典中，其中的键来自数组 B。最后，我们将ix 的值增加l 的值，以确保我们制作的下一个切片通过数组C 向前移动。

我希望这会有所帮助。