如何对三维数组进行筛选

在Python中，如果你有三维数组（通常称为numpy数组），想要筛选特定条件下的元素，可以使用numpy的布尔索引（Boolean Indexing）。以下是一个例子，假设你有一个三维数组`arr`，每个维度代表一个特征（如颜色、尺寸和类别），你想找出所有红色且大小大于某个值的小物体：

import numpy as np

# 假设 arr 是一个三维数组，例如形状为 (color, size, category)
# color, size 和 category 分别是三个轴的索引
arr_shape = arr.shape

# 确定筛选条件
color_filter = arr[:,:,0] == 'red'  # 检查颜色是否为红色
size_filter = arr[:,:,1] > certain_size  # 某个具体的大小值，替换为实际数值

# 使用布尔数组进行过滤
filtered_arr = arr[color_filter & size_filter]

print(filtered_arr)

这里我们先创建了两个布尔数组，分别对应于颜色和尺寸的条件，然后将它们通过按位与(&)操作结合在一起，得到满足两个条件的结果。