如何在 Python 中输入二维数组

在 Python 中，我们可以使用嵌套列表来表示二维数组。但是，这种方法不够直观，而且容易出错。因此，在本文中，我们将讨论如何使用 numpy 库来输入二维数组。

numpy

Numpy 是 Python 中用于科学计算的开源库。它提供了一个高性能的多维数组对象，以及用于处理这些数组的工具。以下是如何在 Python 中使用 numpy 库：

import numpy as np

二维数组的输入

现在，我们来看一下如何在 Python 中输入二维数组。我们可以使用 numpy 库的 array() 函数来创建二维数组。以下是一个例子：

arr = np.array([[1, 2], [3, 4]])

这将创建一个 2 行 2 列的二维数组，其中第一行为 [1, 2]，第二行为 [3, 4]。我们可以使用 print() 函数来打印这个数组：

print(arr)

输出结果为：

[[1 2]
 [3 4]]

输入任意大小的二维数组

如果我们想要输入任意大小的二维数组，我们可以先输入每一行的值，然后使用 vstack() 函数来把它们堆叠起来。以下是一个例子：

# 输入一个 3 行 4 列的二维数组
a = np.array([input().split() for _ in range(3)], dtype=int)
# 打印这个数组
print(a)

在这个例子中，我们首先使用 input() 函数来输入每一行的值，然后使用 split() 函数分离每个值，并将它们转换为整数。最后，我们使用 vstack() 函数把它们都堆叠在一起，创建一个 3 行 4 列的二维数组。

总结

在 Python 中，我们可以使用 numpy 库来创建和处理多维数组。输入二维数组的最简单的方法是使用 array() 函数。如果我们想要输入任意大小的二维数组，我们可以先输入每一行的值，然后使用 vstack() 函数来把它们堆叠起来。