NumPy是Python中用于科学计算和数值运算的常用库,其核心数据结构是ndarray(N-dimensional array),即N维数组,它可以表示矩阵、向量等多维数组。
一、数组的创建
1.可以使用numpy中的array函数从Python列表创建一个数组。例如:
#导入numpy数组
import numpy as pd
#创建一个3*4的矩阵数组
a=pd.array([[1,2,3,4],
[5,6,7,8],
[9,10,11,12]])
print(a)
结果如下:
[[ 1 2 3 4]
[ 5 6 7 8]
[ 9 10 11 12]]
2.使用numpy中的arange函数创建一维数组
可以使用numpy中的arange函数创建一维数组。该函数可以指定起始值、终止值和步长。例如:
import numpy as np
a = np.arange(1, 10, 2)
print(a)
结果如下:
[1 3 5 7 9]
此外,还可以使用numpy中的zeros和ones函数创建数组、使用numpy中的eye和identity函数创建数组、从文件读取数据创建数组等方式构建数组。
二、数组的属性
1. numpy中,ndarray数组的属性shape表示数组的维度大小,即一个元组,其中每个元素表示该维度的大小:
import numpy as np
# 创建一个二维数组
a = np.array([[1, 2, 3], [4, 5, 6]])
# 查看a的维度大小
print(a.shape) # 输出 (2, 3),表示a有2行3列
# 创建一个三维数组
b = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
# 查看b的维度大小
print(b.shape) # 输出 (2, 2, 2),表示b有2个2行2列的矩阵
# 改变数组的形状
c = np.array([1, 2, 3, 4, 5, 6])
print(c.shape) # 输出 (6,),表示c是一个长度为6的一维数组
d = c.reshape((2, 3)) # 将c变为一个2行3列的二维数组
print(d.shape) # 输出 (2, 3),表示d有2行3列
2.在NumPy中,ndarray数组的属性ndim表示数组的维度数量:
import numpy as np
# 创建一个二维数组
a = np.array([[1, 2, 3], [4, 5, 6]])
# 查看a的维度数量
print(a.ndim) # 输出 2,表示a有两个维度
# 创建一个三维数组
b = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
# 查看b的维度数量
print(b.ndim) # 输出 3,表示b有三个维度
3.在NumPy中,ndarray数组的属性dtype表示数组中元素的数据类型,例如整数、浮点数等。属性dtype.name则返回数据类型的字符串名称:
import numpy as np
# 创建一个整数类型的数组
a = np.array([1, 2, 3])
# 查看a中元素的数据类型
print(a.dtype) # 输出 int64,表示a中元素的数据类型是64位整数
# 创建一个浮点数类型的数组
b = np.array([1.0, 2.0, 3.0])
# 查看b中元素的数据类型
print(b.dtype) # 输出 float64,表示b中元素的数据类型是64位浮点数
4.在NumPy中,ndarray数组的属性itemsize表示数组中每个元素的字节大小:
import numpy as np
# 创建一个64位整数类型的数组
a = np.array([1, 2, 3], dtype=np.int64)
# 查看a中每个元素的字节大小
print(a.itemsize) # 输出 8,表示a中每个元素占用8个字节的空间
# 创建一个32位浮点数类型的数组
b = np.array([1.0, 2.0, 3.0], dtype=np.float32)
# 查看b中每个元素的字节大小
print(b.itemsize) # 输出 4,表示b中每个元素占用4个字节的空间
5.在NumPy中,ndarray数组的属性size表示数组中元素的总数:
import numpy as np
# 创建一个2行3列的数组
a = np.array([[1, 2, 3], [4, 5, 6]])
# 查看a中元素的总数
print(a.size) # 输出 6,表示a中有6个元素
# 创建一个3个元素的一维数组
b = np.array([1, 2, 3])
# 查看b中元素的总数
print(b.size) # 输出 3,表示b中有3个元素
6.在Python中,type()函数用于返回给定对象的类型:
import numpy as np
# 创建一个一维数组
a = np.array([1, 2, 3])
# 使用type()函数查看a的类型
print(type(a)) # 输出 <class 'numpy.ndarray'>
# 创建一个二维数组
b = np.array([[1, 2, 3], [4, 5, 6]])
# 使用type()函数查看b的类型
print(type(b)) # 输出 <class 'numpy.ndarray'>