温馨提示×

python中numpy库array函数用法

小亿
128
2023-12-26 02:17:49
栏目: 编程语言

在Python中,numpy库的array函数用于将列表或元组转换为一个numpy数组。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], [7, 8, 9]])
print(arr2)
# [[1 2 3]
#  [4 5 6]
#  [7 8 9]]

# 创建一个带有指定数据类型的数组
arr3 = np.array([1, 2, 3, 4, 5], dtype=float)
print(arr3)  # [1. 2. 3. 4. 5.]

# 通过元组创建数组
arr4 = np.array((1, 2, 3, 4, 5))
print(arr4)  # [1 2 3 4 5]

需要注意的是,numpy数组中的元素类型是相同的,如果输入的列表或元组中包含不同类型的元素,numpy会自动尝试将其转换为相同的类型。

0