python - 如何在 python 中创建一个给定 n 行和 k 列、起始数 x 和步长 y 的数组

Question

假设我要求一些 n 行和 k 列的输入:

rows = n
columns = k

我想要做的是创建一个维度数组(行，列)，然后用从某个数字x开始的一组数字填充所述数组一些步长y，这样无论行和列的输入如何，它都能够创建一个具有“最后一个维度”的数组。 “/数组的右下角元素是一些尚未确定的数字z。

作为示例输入，我可以:

rows = 4
columns = 3

# Starting number
x = 1

# Step size
y = 2

预期的输出将是:

array = [[ 1  3  5]
         [ 7  9 11]
         [13 15 17]
         [19 21 23]]

当数组的维度发生变化时，其中的数字也会发生变化，具体取决于数组的维度、起始数字和步长。

Answer 1

是的，你可以这样做:

import numpy
rows = 4
columns = 3
# Starting number
x = 1
# Step size
y = 2

array = np.arange(x, (rows*columns)*y+x, y).reshape(rows, columns)

请查看https://numpy.org/doc/stable/reference/generated/numpy.arange.html和 https://numpy.org/doc/stable/reference/generated/numpy.reshape.html