Python 获取数组的输入

在 Python 中获取用户输入数组是一个常见的需求，本文将介绍几种不同的方法来获取数组的输入。

方法一：使用 input() 函数

可以使用 input() 函数获取用户输入的数组，如下所示：

arr = []
n = int(input("请输入数组的长度："))
for i in range(n):
    arr.append(int(input(f"请输入第{i+1}个数：")))
print(arr)

在上面的例子中，我们使用一个空列表 arr 来存储输入的整数。首先，使用 int() 函数将输入转换为整数，然后将其添加到 arr 中。

方法二：使用 split() 函数

另一种获取用户输入数组的方法是使用 split() 函数。这种方法要求用户在输入时将元素用空格分隔开。如下所示：

arr = input("请输入数组（用空格隔开）：").split()
arr = [int(i) for i in arr]
print(arr)

在上面的例子中，我们通过调用 input() 函数从用户那里获取输入。接下来，我们使用 split() 函数将输入字符串分割成多个元素，然后使用列表理解 (list comprehension) 将这些元素转换为整数并存储在 arr 中。

方法三：使用 numpy 库

如果您正在处理大量的数值数据，那么 numpy 库提供了更高效的方式来获取用户输入数组。如下所示：

import numpy as np

arr = np.array(input("请输入数组（用空格分隔）：").split(), dtype='int')
print(arr)

在上面的例子中，我们导入了 numpy 库并使用其中的 array() 函数将字符串输入转换为 numpy 数组，然后将其打印输出。

以上是获取数组输入的几种常见方法。根据您的具体需求，选择适合您的方法即可。