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

使用array()函数在Python中创建多维数组

发布时间:2023-12-26 17:37:25

在Python中,可以使用numpy库中的array()函数来创建多维数组。numpy是一个强大的数学库,提供了许多用于处理多维数组的函数和方法。

下面是使用array()函数创建多维数组的示例代码:

import numpy as np

# 创建一维数组
arr1 = np.array([1, 2, 3])
print(arr1)
# 输出: [1 2 3]

# 创建二维数组
arr2 = np.array([[1, 2, 3], [4, 5, 6]])
print(arr2)
# 输出:
# [[1 2 3]
#  [4 5 6]]

# 创建三维数组
arr3 = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
print(arr3)
# 输出:
# [[[1 2]
#   [3 4]]

#  [[5 6]
#   [7 8]]]

# 创建指定数据类型的数组
arr4 = np.array([1, 2, 3], dtype=float)
print(arr4)
# 输出: [1. 2. 3.]

# 创建指定维度的数组
arr5 = np.array([1, 2, 3], ndmin=3)
print(arr5)
# 输出: [[[1 2 3]]]

可以在创建数组时指定数组的数据类型(如float、int等),并通过ndmin参数指定数组的维度。

除了使用array()函数,还可以使用zeros()、ones()、full()、eye()等函数创建多维数组。接下来,我们使用一个例子来说明如何创建和操作多维数组。

# 创建一个3x3的二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 获取数组的维度
print(arr.ndim)
# 输出: 2

# 获取数组的形状
print(arr.shape)
# 输出: (3, 3)

# 获取数组的总大小
print(arr.size)
# 输出: 9

# 获取数组的数据类型
print(arr.dtype)
# 输出: int64

# 访问数组中的元素
print(arr[0, 0])
# 输出: 1

# 切片操作
print(arr[:, 1])
# 输出: [2 5 8]

# 数组运算
arr2 = np.array([[1, 1, 1], [2, 2, 2], [3, 3, 3]])
print(arr + arr2)
# 输出:
# [[ 2  3  4]
#  [ 6  7  8]
#  [10 11 12]]

print(arr * arr2)
# 输出:
# [[ 1  2  3]
#  [ 8 10 12]
#  [21 24 27]]

上面的例子中,我们创建了一个3x3的二维数组arr,并进行了一些常见的操作,包括获取数组的维度、形状、大小、数据类型等,以及对数组进行访问、切片和运算。

总结来说,使用array()函数可以在Python中创建多维数组,numpy库提供了丰富的函数和方法来处理和操作多维数组,对于科学计算和数据分析等领域非常有用。