Python:使用 def 和 for 循环导出列表中一系列值的多个数组

Question

我已经浏览了有关此主题的多个其他线程，但找不到任何适合我需要的内容。我试图达到的最终结果是范围内每个数字的 numpy 数字数组。我尝试过使用 for 循环和 python 的 def 功能来完成此操作，但我觉得我可能错过了一个步骤。到目前为止我编写的代码如下。

source_values = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21]

def CreateSourceList(z):
    y = []
    for i in z:
        Start = 3332
        n = 21
        End = 11733
        y = np.arange(Start + i, End, n)
    return y

代码为 source_values 列表中的每个值创建一个 numpy 数组，并在数组的计算中使用源值。但是，当我使用新函数(CreateSourceList)时，它只会返回 source_values 列表中最后一个源值的 numpy 数组。那么是否有办法为 source_values 列表中的每个源值导出 numpy 数组？

• 对于 source_values = 1，结果应为 [3333, 3354, 3375, 3396, ...]
• 对于 source_values = 2，结果应为 [3334, 3355, 3376, 3397, ...]
• 对于 source_values = 3，结果应为 [3335, 3356, 3377, 3398, ...]

谢谢!

Answer 1

您可以仅使用列表理解来生成数字。另外，只需将偏移量 i 添加到原始范围即可。

Start = 3332
n = 21
End = 3395
source_values = [1, 2, 3]

>>> [np.arange(Start, End, n) + i for i in source_values]
[array([3333, 3354, 3375]),
 array([3334, 3355, 3376]),
 array([3335, 3356, 3377])]