如何根据另一列中的条件找到3D数组中列的最大值？

import numpy as np

# 生成一个3D数组
arr = 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], [25, 26, 27]]])

# 要根据的条件所在的列
condition_column = 1

# 初始化最大值为负无穷
max_value = float('-inf')

# 遍历每个矩阵
for matrix in arr:
    # 遍历每一行
    for row in matrix:
        # 获取条件列的值
        value = row[condition_column]
        # 更新最大值
        if value > max_value:
            max_value = value

print("满足条件的列的最大值为:", max_value)