NumPy
是 Python 的一个开源数值计算扩展库,用于高性能的科学计算和数据分析。它提供了一个强大的N维数组对象ndarray
,以及用于对这些数组进行操作的各种函数。
通过pip
安装:
pip install numpy
创建数组
从列表创建
numpy.array
可以通过多种方式创建,最直接的方式是使用np.array
函数,将Python列表或元组转换为ndarray
对象。
import numpy as np
# 从列表创建数组
arr_from_list = np.array([1, 2, 3, 4, 5])
print(arr_from_list)
# 输出:[1 2 3 4 5]
# 从嵌套列表创建二维数组
arr_from_nested_list = np.array([[1, 2, 3], [4, 5, 6]])
print(arr_from_nested_list)
# 输出:
# [[1 2 3]
# [4 5 6]]
在创建numpy.array
时,可以通过dtype
参数指定数组元素的数据类型。NumPy
提供了丰富的数据类型,包括int
、float
、complex
、bool
、object
等。
arr_int = np.array([1, 2, 3], dtype=np.int32)
arr_float = np.array([1, 2, 3], dtype=np.float64)
print("整型数组:", arr_int