这里是文章模块栏目内容页
python如何操作多维数组

在Python中,可以使用NumPy库来操作多维数组,以下是一些常用的操作:

python如何操作多维数组
(图片来源网络,侵删)

1、导入NumPy库

import numpy as np

2、创建多维数组

创建一个3x4的二维数组
arr = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])
创建一个2x3x4的三维数组
arr_3d = np.array([[[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]],
                   [[13, 14, 15, 16], [17, 18, 19, 20], [21, 22, 23, 24]]])

3、访问多维数组的元素

访问二维数组的第一个元素
first_element = arr[0, 0]
访问三维数组的第一个元素的第一个子数组的第二个元素
third_element = arr_3d[0, 0, 1]

4、修改多维数组的元素

修改二维数组的第一个元素为100
arr[0, 0] = 100
修改三维数组的第一个元素的第一个子数组的第二个元素为200
arr_3d[0, 0, 1] = 200

5、获取多维数组的形状(维度)和大小

获取二维数组的形状(维度)和大小
shape_2d = arr.shape
size_2d = arr.size
获取三维数组的形状(维度)和大小
shape_3d = arr_3d.shape
size_3d = arr_3d.size

6、对多维数组进行切片操作

对二维数组进行切片操作,获取第一行和第二列的元素
slice_2d = arr[:2, 1:]
对三维数组进行切片操作,获取第一个元素的所有子数组的前两个元素
slice_3d = arr_3d[0, :2, :]

7、对多维数组进行拼接、分割等操作

对两个二维数组进行垂直拼接
concat_arr = np.vstack((arr, arr))
对两个三维数组进行水平拼接
concat_arr_3d = np.hstack((arr_3d, arr_3d))
对一个三维数组沿着第三个轴(深度)进行分割,得到两个子数组
split_arr_3d = np.split(arr_3d, 2, axis=2)
更多栏目