📜  如何在 python 中创建数组(1)

📅  最后修改于: 2023-12-03 15:38:23.639000             🧑  作者: Mango

在 Python 中创建数组

数组在计算机科学中是一种非常常见的数据结构。在 Python 中,我们可以使用内置的 array 模块或者 numpy 模块来创建数组。下面是介绍如何使用这两种方法创建数组的详细内容。

使用 array 模块创建数组

首先,我们需要导入 array 模块。

import array

然后,我们可以使用 array.array() 函数来创建数组。这个函数有两个参数:数组的类型和数组的初始值。

# 创建一个包含整数的数组
my_array = array.array('i', [1, 2, 3, 4, 5])

# 创建一个包含浮点数的数组
my_array = array.array('d', [1.0, 2.0, 3.0, 4.0, 5.0])

在这个例子中,我们分别创建了一个包含整数和浮点数的数组。注意,array 模块中支持的类型包括整数、浮点数、双精度浮点数等。具体的类型码请参考官方文档

除了上面的方法之外,我们还可以使用其他函数来创建数组。

# 创建一个空的数组
my_array = array.array('i')

# 创建一个包含 n 个元素的数组,初始值都为 0
my_array = array.array('i', [0] * n)

# 创建一个包含 n 个元素的数组,初始值都为 1
my_array = array.array('i', [1] * n)
使用 numpy 模块创建数组

除了 array 模块,我们还可以使用更加强大的 numpy 模块来创建数组。我们需要先导入这个模块。

import numpy as np

然后,我们就可以使用 numpy.array() 函数来创建数组。

# 创建一个包含整数的数组
my_array = np.array([1, 2, 3, 4, 5])

# 创建一个包含浮点数的数组
my_array = np.array([1.0, 2.0, 3.0, 4.0, 5.0])

除了上面的方法之外,我们还可以使用其他函数来创建数组。

# 使用 numpy.zeros() 函数创建一个初始值全为 0 的数组
my_array = np.zeros(n)

# 使用 numpy.ones() 函数创建一个初始值全为 1 的数组
my_array = np.ones(n)

# 使用 numpy.arange() 函数创建一个范围数组
my_array = np.arange(start, stop, step)

到此为止,我们就介绍了如何使用 array 模块和 numpy 模块来创建数组。两种方法各有优缺点,需要根据实际情况进行选择。