欢迎访问宙启技术站
智能推送

学习使用Numpy核心数字模块的基本方法

发布时间:2023-12-27 14:34:53

Numpy是一个强大的Python库,用于运算和处理大规模多维数组以及矩阵计算。在Numpy的核心数字模块中,有许多基本的方法可以用来操作和处理数组。本文将介绍一些常用的方法,并提供相应的使用例子。

1. 创建数组

Numpy提供了多种创建数组的方法,例如使用np.array函数,可以将列表或元组转换为数组:

import numpy as np

arr1 = np.array([1, 2, 3, 4, 5])
print(arr1)  # [1 2 3 4 5]

arr2 = np.array([[1, 2, 3], [4, 5, 6]])
print(arr2)
'''
[[1 2 3]
 [4 5 6]]
'''

arr3 = np.zeros((3, 3))
print(arr3)
'''
[[0. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]]
'''

2. 数组的属性

数组的属性包括形状、维度、元素类型等,可以使用shapendimdtype属性来获取:

import numpy as np

arr1 = np.array([1, 2, 3, 4, 5])
print(arr1.shape)  # (5,)
print(arr1.ndim)  # 1
print(arr1.dtype)  # int64

arr2 = np.array([[1, 2, 3], [4, 5, 6]])
print(arr2.shape)  # (2, 3)
print(arr2.ndim)  # 2
print(arr2.dtype)  # int64

3. 数组的索引和切片

可以使用索引来访问数组的特定元素,并使用切片来获取数组的子数组。索引的起始值为0,并支持负数索引表示从末尾开始计数:

import numpy as np

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

print(arr[0])  # 1
print(arr[-1])  # 5

sub_arr = arr[1:3]
print(sub_arr)  # [2 3]

arr[0] = 0
print(arr)  # [0 2 3 4 5]

4. 数组的运算

Numpy数组支持基本的算术运算,如加减乘除,并且可以直接对整个数组进行运算,也可以对数组的每个元素进行运算:

import numpy as np

arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([2, 3, 4, 5, 6])

# 数组之间的运算
arr3 = arr1 + arr2
print(arr3)  # [3 5 7 9 11]

arr4 = arr1 * arr2
print(arr4)  # [ 2  6 12 20 30]

# 数组与常数之间的运算
arr5 = arr1 + 1
print(arr5)  # [2 3 4 5 6]

arr6 = arr2 ** 2
print(arr6)  # [ 4  9 16 25 36]

5. 数组的函数操作

Numpy提供了多种函数操作,例如计算数组的平均值、最大值、最小值等:

import numpy as np

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

print(np.mean(arr))  # 3.0
print(np.max(arr))  # 5
print(np.min(arr))  # 1
print(np.sum(arr))  # 15

在本文中,我们介绍了Numpy核心数字模块的一些基本方法,并提供了相应的使用例子。这些方法包括创建数组、数组的属性、数组的索引和切片、数组的运算以及数组的函数操作。这些方法将帮助你更好地理解和使用Numpy库来处理和运算数组。