在任意轴上迭代体积的更多pythonic方式？

arr = [[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]]

# 在x轴上迭代体积
x_volumes = [volume for y in arr for x in y for volume in x]
print(x_volumes)

# 在y轴上迭代体积
y_volumes = [volume for y in arr for volume in y]
print(y_volumes)

# 在z轴上迭代体积
z_volumes = [volume for volume in arr]
print(z_volumes)

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
[[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]]
[[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]]