Python | numpy matrix.astype()

NumPy是Python中用于科学计算的库之一。NumPy支持进行多维数组（数组、矩阵等）的运算，提供了大量的函数库以便于我们进行科学计算。matrix.astype()是NumPy中用于改变矩阵数据类型的函数。

语法

matrix.astype(dtype)

参数说明

• dtype：指定矩阵的数据类型。

返回值

返回更改数据类型后的矩阵。

示例

以下是使用matrix.astype()函数的示例：

import numpy as np

# 创建一个整型矩阵
a = np.matrix('1 2; 3 4')
print("原矩阵数据类型：", a.dtype)

# 将矩阵数据类型改为 float
b = a.astype('float')
print("更改数据类型后的矩阵：\n", b)
print("更改数据类型后的矩阵数据类型：", b.dtype)

# 将矩阵数据类型改为 complex
c = a.astype('complex')
print("更改数据类型后的矩阵：\n", c)
print("更改数据类型后的矩阵数据类型：", c.dtype)

输出结果如下：

原矩阵数据类型： int32
更改数据类型后的矩阵：
 [[1. 2.]
 [3. 4.]]
更改数据类型后的矩阵数据类型： float64
更改数据类型后的矩阵：
 [[1.+0.j 2.+0.j]
 [3.+0.j 4.+0.j]]
更改数据类型后的矩阵数据类型： complex128

在上面的示例中，我们先创建了一个整型矩阵a，然后使用a.astype()函数将矩阵a的数据类型分别更改为float和complex类型，并输出更改数据类型后的矩阵及其数据类型。从结果中可以看出，使用astype()函数可以很方便地改变矩阵的数据类型。