Python: экспорт нескольких массивов с использованием циклов def и for для диапазона значений в списке

Question

Я просмотрел несколько других тем на эту тему, но не смог найти ничего подходящего для того, что мне нужно.Конечным результатом, которого я пытаюсь достичь, является массив значений для каждого числа в диапазоне.Я попытался использовать для циклов и функциональности 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 ​​и использует значение источника при вычислении массива.Тем не менее, когда я использую новую функцию (CreateSourceList), она будет возвращать только пустой массив для последнего значения источника в списке source_values.Так есть ли в любом случае экспортировать массив NumPy для каждого значения источника в списке source_values?

• Для 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])]