📌  相关文章
📜  创建数组时指定 dtype - Python (1)

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

创建数组时指定 dtype - Python

当我们创建一个数组时,我们可以指定它的数据类型(dtype)。指定数组的dtype可以让我们更加精确地控制它的内存分配和数据存储方式。

在python中,我们通常使用numpy库来创建数组。numpy数组有很多数据类型可以选择,包括整数、浮点数、布尔值和字符串等。

以下示例展示如何使用numpy创建一个包含3个浮点数的数组,其中每个元素占用8个字节的内存空间:

import numpy as np

a = np.array([1.0, 2.0, 3.0], dtype=np.float64)
print(a)

输出结果为:

array([1., 2., 3.])

其中,np.float64表示数组元素的数据类型为64位的浮点数。

如果我们想把数组中的元素转换成其他数据类型,可以使用numpy的astype()方法。

以下示例展示如何把上面创建的数组转换成整数类型:

b = a.astype(np.int32)
print(b)

输出结果为:

array([1, 2, 3], dtype=int32)

其中,np.int32表示数组元素的数据类型为32位的有符号整数。

我们也可以使用numpy的dtype属性来查看数组的数据类型:

print(a.dtype)

输出结果为:

float64

总之,numpy的dtype让我们可以在创建和转换数组时更加精确地控制数据类型和内存分配。这对于某些高性能计算任务和数据分析任务来说尤为重要。