如何在 python 中以单行方式获取数组输入

在 python 中，我们可以使用 input() 函数来获取用户输入，但是如果要获取一个数组的输入，需要使用多行代码来完成。不过，我们可以通过一些技巧，在 python 中以单行方式获取数组输入。这里介绍两种方法。

利用 split() 函数

首先，我们可以利用字符串的 split() 函数来将用户输入的一行字符串分割成数组。具体代码如下：

a = input().split()

这里，a 是一个数组，输入的数据以空格分隔。例如，如果用户输入了 1 2 3 4 5，则 a 的值为 ['1', '2', '3', '4', '5']。

使用列表推导式

另一种方法是使用列表推导式来构建数组。具体代码如下：

a = [int(x) for x in input().split()]

这里，和上面一样，我们先利用 split() 函数将用户输入的字符串分割成数组。不过这里我们使用列表推导式来将每个字符串转换成整数。例如，如果用户输入了 1 2 3 4 5，则 a 的值为 [1, 2, 3, 4, 5]。

这两种方法都可以以单行方式获取数组输入，在一些简单场景中都可以使用。

参考资料：How to Get Python Input as a List in a Single Line