python 的三维数组 运算

在Python中，可以使用NumPy库来创建和操作三维数组。NumPy是一个强大的数学计算库，它提供了一种称为ndarray的数据结构，它可以容纳任意维度的数据，包括三维数组。

创建三维数组的基本语法如下：

import numpy as np

# 创建一个全零的三维数组
arr_3d = np.zeros((rows, cols, depth))

# 或者通过列表生成器创建
arr_3d = np.array([[[0 for _ in range(cols)] for _ in range(rows)] for _ in range(depth)])

# 初始化特定值
arr_3d = np.full((rows, cols, depth), value)

# 从现有数据创建三维数组
arr_3d = np.fromfunction(lambda i, j, k: i * j + k, shape=(rows, cols, depth))

三维数组支持各种运算，如元素级别的加减乘除、矩阵运算（如点积、转置等）、统计函数（求和、平均值等），以及广播功能，这使得处理高维数据变得非常方便。例如：

# 加法运算
arr1 + arr2

# 平均值计算
np.mean(arr_3d, axis=某个轴)

# 点积
np.dot(arr1, arr2)